Ste*_*nko 5 hadoop recommendation-engine apache-spark-mllib
我正在使用这段代码来计算建议:
SparkSession spark = SparkSession
.builder()
.appName("SomeAppName")
.config("spark.master", "local")
.getOrCreate();
JavaRDD<Rating> ratingsRDD = spark
.read().textFile(args[0]).javaRDD()
.map(Rating::parseRating);
Dataset<Row> ratings = spark.createDataFrame(ratingsRDD, Rating.class);
ALS als = new ALS()
.setMaxIter(1)
.setRegParam(0.01)
.setUserCol("userId")
.setItemCol("movieId")
.setRatingCol("rating");
ALSModel model = als.fit(ratings);
model.setColdStartStrategy("drop");
Dataset<Row> rowDataset = model.recommendForAllUsers(50);
Run Code Online (Sandbox Code Playgroud)
我想跟踪迭代的进度,理想情况下可以查看迭代编号,是否可以为此提供Java回调?
| 归档时间: |
|
| 查看次数: |
48 次 |
| 最近记录: |