package com.intel.daal.examples.optimization_solvers;
import com.intel.daal.algorithms.optimization_solver.lbfgs.*;
import com.intel.daal.algorithms.optimization_solver.iterative_solver.Input;
import com.intel.daal.algorithms.optimization_solver.iterative_solver.InputId;
import com.intel.daal.algorithms.optimization_solver.iterative_solver.OptionalInputId;
import com.intel.daal.algorithms.optimization_solver.iterative_solver.Result;
import com.intel.daal.algorithms.optimization_solver.iterative_solver.ResultId;
import com.intel.daal.algorithms.optimization_solver.iterative_solver.OptionalResultId;
import com.intel.daal.algorithms.OptionalArgument;
import com.intel.daal.data_management.data.HomogenNumericTable;
import com.intel.daal.data_management.data.MergedNumericTable;
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 LBFGSOptResDenseBatch {
private static final long nFeatures = 10;
private static final long nIterations = 1000;
private static final double stepLength = 1.0e-4;
private static double[] initialPoint = {100, 100, 100, 100, 100, 100, 100, 100, 100, 100, 100};
private static double[] expectedPoint = { 11, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10};
private static final String datasetFileName = "../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, datasetFileName,
DataSource.DictionaryCreationFlag.DoDictionaryFromContext,
DataSource.NumericTableAllocationFlag.NotAllocateNumericTable);
NumericTable data = new HomogenNumericTable(context, Float.class, nFeatures, 0, NumericTable.AllocationFlag.DoNotAllocate);
NumericTable dependentVariables = new HomogenNumericTable(context, Float.class, 1, 0,
NumericTable.AllocationFlag.DoNotAllocate);
MergedNumericTable mergedData = new MergedNumericTable(context);
mergedData.addNumericTable(data);
mergedData.addNumericTable(dependentVariables);
dataSource.loadDataBlock(mergedData);
com.intel.daal.algorithms.optimization_solver.mse.Batch mseObjectiveFunction =
new com.intel.daal.algorithms.optimization_solver.mse.Batch(context, Float.class,
com.intel.daal.algorithms.optimization_solver.mse.Method.defaultDense, data.getNumberOfRows());
mseObjectiveFunction.getInput().set(com.intel.daal.algorithms.optimization_solver.mse.InputId.data, data);
mseObjectiveFunction.getInput().set(com.intel.daal.algorithms.optimization_solver.mse.InputId.dependentVariables,
dependentVariables);
Batch algorithm = new Batch(context, Float.class, Method.defaultDense);
algorithm.parameter.setFunction(mseObjectiveFunction);
algorithm.parameter.setNIterations(nIterations/2);
algorithm.parameter.setStepLengthSequence(
new HomogenNumericTable(context, Float.class, 1, 1, NumericTable.AllocationFlag.DoAllocate, stepLength));
algorithm.parameter.setOptionalResultRequired(true);
algorithm.input.set(InputId.inputArgument, new HomogenNumericTable(context, initialPoint, 1, nFeatures + 1));
Result result = algorithm.compute();
Service.printNumericTable("Resulting coefficients after first compute():", result.get(ResultId.minimum));
Service.printNumericTable("Number of iterations performed:", result.get(ResultId.nIterations));
algorithm.input.set(InputId.inputArgument, result.get(ResultId.minimum));
algorithm.input.set(OptionalInputId.optionalArgument, result.get(OptionalResultId.optionalResult));
result = algorithm.compute();
NumericTable expected = new HomogenNumericTable(context, expectedPoint, 1, nFeatures + 1);
Service.printNumericTable("Expected coefficients:", expected);
Service.printNumericTable("Resulting coefficients after second compute():", result.get(ResultId.minimum));
Service.printNumericTable("Number of iterations performed:", result.get(ResultId.nIterations));
}
}