47 from daal.algorithms
import em_gmm
48 from daal.data_management
import FileDataSource, DataSourceIface
50 utils_folder = os.path.realpath(os.path.abspath(os.path.dirname(os.path.dirname(__file__))))
51 if utils_folder
not in sys.path:
52 sys.path.insert(0, utils_folder)
53 from utils
import printNumericTable
55 DAAL_PREFIX = os.path.join(
'..',
'data')
58 datasetFileName = os.path.join(DAAL_PREFIX,
'batch',
'em_gmm.csv')
61 if __name__ ==
"__main__":
64 dataSource = FileDataSource(
66 DataSourceIface.doAllocateNumericTable,
67 DataSourceIface.doDictionaryFromContext
69 nFeatures = dataSource.getNumberOfColumns()
72 dataSource.loadDataBlock()
76 initAlgorithm = em_gmm.init.Batch(nComponents)
79 initAlgorithm.input.set(em_gmm.init.data, dataSource.getNumericTable())
82 resultInit = initAlgorithm.compute()
85 algorithm = em_gmm.Batch(nComponents)
88 algorithm.input.setTable(em_gmm.data, dataSource.getNumericTable())
89 algorithm.input.setValues(em_gmm.inputValues, resultInit)
92 result = algorithm.compute()
95 printNumericTable(result.getResult(em_gmm.weights),
"Weights")
96 printNumericTable(result.getResult(em_gmm.means),
"Means")
97 for i
in range(nComponents):
98 printNumericTable(result.getCovariances(em_gmm.covariances, i),
"Covariance")