31 import daal.algorithms.optimization_solver
as optimization_solver
32 import daal.algorithms.optimization_solver.mse
33 import daal.algorithms.optimization_solver.adagrad
34 import daal.algorithms.optimization_solver.iterative_solver
36 from daal.data_management
import (
37 DataSourceIface, FileDataSource, HomogenNumericTable, MergedNumericTable, NumericTableIface
40 utils_folder = os.path.realpath(os.path.abspath(os.path.dirname(os.path.dirname(__file__))))
41 if utils_folder
not in sys.path:
42 sys.path.insert(0, utils_folder)
43 from utils
import printNumericTable
45 datasetFileName = os.path.join(
'..',
'data',
'batch',
'mse.csv')
50 accuracyThreshold = 0.0000001
53 startPoint = np.array([[8], [2], [1], [4]], dtype=np.float64)
55 if __name__ ==
"__main__":
58 dataSource = FileDataSource(datasetFileName,
59 DataSourceIface.notAllocateNumericTable,
60 DataSourceIface.doDictionaryFromContext)
63 data = HomogenNumericTable(nFeatures, 0, NumericTableIface.doNotAllocate)
64 dependentVariables = HomogenNumericTable(1, 0, NumericTableIface.doNotAllocate)
65 mergedData = MergedNumericTable(data, dependentVariables)
68 dataSource.loadDataBlock(mergedData)
70 nVectors = data.getNumberOfRows()
72 mseObjectiveFunction = optimization_solver.mse.Batch(nVectors)
73 mseObjectiveFunction.input.set(optimization_solver.mse.data, data)
74 mseObjectiveFunction.input.set(optimization_solver.mse.dependentVariables, dependentVariables)
77 adagradAlgorithm = optimization_solver.adagrad.Batch(mseObjectiveFunction)
80 adagradAlgorithm.input.setInput(optimization_solver.iterative_solver.inputArgument, HomogenNumericTable(startPoint))
81 adagradAlgorithm.parameter.learningRate = HomogenNumericTable(1, 1, NumericTableIface.doAllocate, learningRate)
82 adagradAlgorithm.parameter.nIterations = nIterations
83 adagradAlgorithm.parameter.accuracyThreshold = accuracyThreshold
84 adagradAlgorithm.parameter.batchSize = batchSize
88 res = adagradAlgorithm.compute()
91 printNumericTable(res.getResult(optimization_solver.iterative_solver.minimum),
"Minimum:")
92 printNumericTable(res.getResult(optimization_solver.iterative_solver.nIterations),
"Number of iterations performed:")