48 from daal.algorithms
import em_gmm
49 from daal.data_management
import FileDataSource, DataSourceIface
51 utils_folder = os.path.realpath(os.path.abspath(os.path.dirname(os.path.dirname(__file__))))
52 if utils_folder
not in sys.path:
53 sys.path.insert(0, utils_folder)
54 from utils
import printNumericTable
56 DAAL_PREFIX = os.path.join(
'..',
'data')
59 datasetFileName = os.path.join(DAAL_PREFIX,
'batch',
'em_gmm.csv')
62 if __name__ ==
"__main__":
65 dataSource = FileDataSource(
67 DataSourceIface.doAllocateNumericTable,
68 DataSourceIface.doDictionaryFromContext
70 nFeatures = dataSource.getNumberOfColumns()
73 dataSource.loadDataBlock()
77 initAlgorithm = em_gmm.init.Batch(nComponents)
80 initAlgorithm.input.set(em_gmm.init.data, dataSource.getNumericTable())
83 resultInit = initAlgorithm.compute()
86 algorithm = em_gmm.Batch(nComponents)
89 algorithm.input.setTable(em_gmm.data, dataSource.getNumericTable())
90 algorithm.input.setValues(em_gmm.inputValues, resultInit)
93 result = algorithm.compute()
96 printNumericTable(result.getResult(em_gmm.weights),
"Weights")
97 printNumericTable(result.getResult(em_gmm.means),
"Means")
98 for i
in range(nComponents):
99 printNumericTable(result.getCovariances(em_gmm.covariances, i),
"Covariance")