Examples with AlsRateRecommBatchOp com.alibaba.alink.operator.batch.recommendation.AlsRateRecommBatchOp used on opensource projects

Example 1 with AlsRateRecommBatchOp

use of com.alibaba.alink.operator.batch.recommendation.AlsRateRecommBatchOp in project Alink by alibaba.

the class ALSExample method main.

public static void main(String[] args) throws Exception {
    String url = "https://alink-release.oss-cn-beijing.aliyuncs.com/data-files/movielens_ratings.csv";
    String schema = "userid bigint, movieid bigint, rating double, timestamp string";
    BatchOperator data = new CsvSourceBatchOp().setFilePath(url).setSchemaStr(schema);
    SplitBatchOp spliter = new SplitBatchOp().setFraction(0.8);
    spliter.linkFrom(data);
    BatchOperator trainData = spliter;
    BatchOperator testData = spliter.getSideOutput(0);
    AlsTrainBatchOp als = new AlsTrainBatchOp().setUserCol("userid").setItemCol("movieid").setRateCol("rating").setNumIter(10).setRank(10).setLambda(0.1);
    BatchOperator model = als.linkFrom(trainData);
    AlsRateRecommBatchOp predictor = new AlsRateRecommBatchOp().setUserCol("userid").setItemCol("movieid").setRecommCol("prediction_result");
    BatchOperator preditionResult = predictor.linkFrom(model, testData).select("rating, prediction_result");
    preditionResult.print();
}