22 import daal.algorithms.implicit_als.training.init
23 import daal.algorithms.implicit_als.prediction.ratings
24 from daal.algorithms.implicit_als
import training, prediction
25 from daal.data_management
import FileDataSource, DataSourceIface
27 utils_folder = os.path.realpath(os.path.abspath(os.path.dirname(os.path.dirname(__file__))))
28 if utils_folder
not in sys.path:
29 sys.path.insert(0, utils_folder)
30 from utils
import printNumericTable
32 DAAL_PREFIX = os.path.join(
'..',
'data')
35 trainDatasetFileName = os.path.join(DAAL_PREFIX,
'batch',
'implicit_als_dense.csv')
45 def initializeModel():
46 global dataTable, initialModel
49 dataSource = FileDataSource(
50 trainDatasetFileName, DataSourceIface.doAllocateNumericTable,
51 DataSourceIface.doDictionaryFromContext
55 dataSource.loadDataBlock()
57 dataTable = dataSource.getNumericTable()
59 initAlgorithm = training.init.Batch()
60 initAlgorithm.parameter.nFactors = nFactors
63 initAlgorithm.input.set(training.init.data, dataTable)
64 res = initAlgorithm.compute()
67 initialModel = res.get(training.init.model)
74 algorithm = training.Batch()
77 algorithm.input.setTable(training.data, dataTable)
78 algorithm.input.setModel(training.inputModel, initialModel)
80 algorithm.parameter.nFactors = nFactors
83 trainingResult = algorithm.compute()
89 algorithm = prediction.ratings.Batch()
90 algorithm.parameter.nFactors = nFactors
92 algorithm.input.set(prediction.ratings.model, trainingResult.get(training.model))
94 res = algorithm.compute()
95 predictedRatings = res.get(prediction.ratings.prediction)
97 printNumericTable(predictedRatings,
"Predicted ratings:")
99 if __name__ ==
"__main__":