package com.intel.daal.examples.optimization_solvers;
import com.intel.daal.algorithms.optimization_solver.saga.*;
import com.intel.daal.algorithms.optimization_solver.iterative_solver.InputId;
import com.intel.daal.algorithms.optimization_solver.iterative_solver.Result;
import com.intel.daal.algorithms.optimization_solver.iterative_solver.ResultId;
import com.intel.daal.data_management.data.HomogenNumericTable;
import com.intel.daal.data_management.data.NumericTable;
import com.intel.daal.data_management.data.MergedNumericTable;
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 SAGADenseBatch {
private static final long nFeatures = 10;
private static final double accuracyThreshold = 0.00000001;
private static final long nIterations = 1000000;
private static double[] initialPoint = { 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0};
private static final String dataFileName = "../data/batch/lbfgs.csv";
private static DaalContext context = new DaalContext();
public static void main(String[] args) throws java.io.FileNotFoundException, java.io.IOException {
FileDataSource dataSource = new FileDataSource(context, dataFileName,
DataSource.DictionaryCreationFlag.DoDictionaryFromContext,
DataSource.NumericTableAllocationFlag.NotAllocateNumericTable);
NumericTable data = new HomogenNumericTable(context, Float.class, nFeatures, 0, NumericTable.AllocationFlag.DoNotAllocate);
NumericTable dataDependents = new HomogenNumericTable(context, Float.class, 1, 0, NumericTable.AllocationFlag.DoNotAllocate);
MergedNumericTable mergedData = new MergedNumericTable(context);
mergedData.addNumericTable(data);
mergedData.addNumericTable(dataDependents);
dataSource.loadDataBlock(mergedData);
com.intel.daal.algorithms.optimization_solver.mse.Batch mseFunction =
new com.intel.daal.algorithms.optimization_solver.mse.Batch(context, Float.class,
com.intel.daal.algorithms.optimization_solver.mse.Method.defaultDense, data.getNumberOfRows());
mseFunction.getInput().set(com.intel.daal.algorithms.optimization_solver.mse.InputId.data, data);
mseFunction.getInput().set(com.intel.daal.algorithms.optimization_solver.mse.InputId.dependentVariables, dataDependents);
Batch sagaAlgorithm = new Batch(context, Float.class, Method.defaultDense);
sagaAlgorithm.parameter.setFunction(mseFunction);
sagaAlgorithm.parameter.setNIterations(nIterations);
sagaAlgorithm.parameter.setAccuracyThreshold(accuracyThreshold);
sagaAlgorithm.input.set(InputId.inputArgument, new HomogenNumericTable(context, initialPoint, 1, nFeatures + 1));
Result result = sagaAlgorithm.compute();
Service.printNumericTable("Minimum:", result.get(ResultId.minimum));
Service.printNumericTable("nIterations:", result.get(ResultId.nIterations));
com.intel.daal.algorithms.optimization_solver.mse.Batch mseFunction_check =
new com.intel.daal.algorithms.optimization_solver.mse.Batch(context, Float.class,
com.intel.daal.algorithms.optimization_solver.mse.Method.defaultDense, data.getNumberOfRows());
mseFunction_check.getInput().set(com.intel.daal.algorithms.optimization_solver.mse.InputId.data, data);
mseFunction_check.getInput().set(com.intel.daal.algorithms.optimization_solver.mse.InputId.dependentVariables, dataDependents);
mseFunction_check.getInput().set(com.intel.daal.algorithms.optimization_solver.mse.InputId.argument, result.get(ResultId.minimum));
mseFunction_check.parameter.setResultsToCompute(com.intel.daal.algorithms.optimization_solver.objective_function.ResultsToComputeId.value);
mseFunction_check.compute();
Service.printNumericTable("value DAAL:", mseFunction_check.getResult().get(com.intel.daal.algorithms.optimization_solver.objective_function.ResultId.valueIdx));
context.dispose();
}
}