22 from daal
import step1Local, step2Master
23 from daal.algorithms
import covariance
24 from daal.data_management
import FileDataSource, DataSourceIface
26 utils_folder = os.path.realpath(os.path.abspath(os.path.dirname(os.path.dirname(__file__))))
27 if utils_folder
not in sys.path:
28 sys.path.insert(0, utils_folder)
29 from utils
import printNumericTable
31 DAAL_PREFIX = os.path.join(
'..',
'data')
37 os.path.join(DAAL_PREFIX,
'distributed',
'covcormoments_dense_1.csv'),
38 os.path.join(DAAL_PREFIX,
'distributed',
'covcormoments_dense_2.csv'),
39 os.path.join(DAAL_PREFIX,
'distributed',
'covcormoments_dense_3.csv'),
40 os.path.join(DAAL_PREFIX,
'distributed',
'covcormoments_dense_4.csv')
43 partialResult = [0] * nBlocks
47 def computestep1Local(block):
51 dataSource = FileDataSource(
52 datasetFileNames[block], DataSourceIface.doAllocateNumericTable,
53 DataSourceIface.doDictionaryFromContext
57 dataSource.loadDataBlock()
60 algorithm = covariance.Distributed(step1Local)
63 algorithm.input.set(covariance.data, dataSource.getNumericTable())
66 partialResult[block] = algorithm.compute()
69 def computeOnMasterNode():
73 algorithm = covariance.Distributed(step2Master)
76 for i
in range(nBlocks):
77 algorithm.input.add(covariance.partialResults, partialResult[i])
83 result = algorithm.finalizeCompute()
85 if __name__ ==
"__main__":
87 for i
in range(nBlocks):
92 printNumericTable(result.get(covariance.covariance),
"Covariance matrix:")
93 printNumericTable(result.get(covariance.mean),
"Mean vector:")