package com.intel.daal.examples.pca;
import com.intel.daal.algorithms.PartialResult;
import com.intel.daal.algorithms.pca.DistributedStep1Local;
import com.intel.daal.algorithms.pca.DistributedStep2Master;
import com.intel.daal.algorithms.pca.InputId;
import com.intel.daal.algorithms.pca.MasterInputId;
import com.intel.daal.algorithms.pca.Method;
import com.intel.daal.algorithms.pca.Result;
import com.intel.daal.algorithms.pca.ResultId;
import com.intel.daal.data_management.data.NumericTable;
import com.intel.daal.data_management.data_source.DataSource;
import com.intel.daal.data_management.data_source.FileDataSource;
import com.intel.daal.examples.utils.Service;
import com.intel.daal.services.DaalContext;
class PCASVDDenseDistributed {
private static final String[] dataset = { "../data/distributed/pca_normalized_1.csv",
"../data/distributed/pca_normalized_2.csv", "../data/distributed/pca_normalized_3.csv",
"../data/distributed/pca_normalized_4.csv",
};
private static final int nNodes = 4;
private static PartialResult[] pres = new PartialResult[nNodes];
private static DaalContext context = new DaalContext();
public static void main(String[] args) throws java.io.FileNotFoundException, java.io.IOException {
for (int i = 0; i < nNodes; i++) {
FileDataSource dataSource = new FileDataSource(context, dataset[i],
DataSource.DictionaryCreationFlag.DoDictionaryFromContext,
DataSource.NumericTableAllocationFlag.DoAllocateNumericTable);
dataSource.loadDataBlock();
DistributedStep1Local pcaLocal = new DistributedStep1Local(context, Double.class, Method.svdDense);
NumericTable data = dataSource.getNumericTable();
pcaLocal.input.set(InputId.data, data);
pres[i] = pcaLocal.compute();
}
DistributedStep2Master pcaMaster = new DistributedStep2Master(context, Double.class, Method.svdDense);
for (int i = 0; i < nNodes; i++) {
pcaMaster.input.add(MasterInputId.partialResults, pres[i]);
}
pcaMaster.compute();
Result res = pcaMaster.finalizeCompute();
NumericTable eigenValues = res.get(ResultId.eigenValues);
NumericTable eigenVectors = res.get(ResultId.eigenVectors);
Service.printNumericTable("Eigenvalues:", eigenValues);
Service.printNumericTable("Eigenvectors:", eigenVectors);
context.dispose();
}
}