Formatting
This commit is contained in:
parent
6586c5e28b
commit
39684eafe3
|
@ -365,7 +365,7 @@ object ALS {
|
|||
|
||||
def main(args: Array[String]) {
|
||||
if (args.length != 5) {
|
||||
println("Usage: ALS <master> <ratings_file> <rank> <num_iters> <output_dir>")
|
||||
println("Usage: ALS <master> <ratings_file> <rank> <iterations> <output_dir>")
|
||||
System.exit(1)
|
||||
}
|
||||
val (master, ratingsFile, rank, iters, outputDir) =
|
||||
|
|
|
@ -9,6 +9,7 @@ class MatrixFactorizationModel(
|
|||
val rank: Int,
|
||||
val userFeatures: RDD[(Int, Array[Double])],
|
||||
val productFeatures: RDD[(Int, Array[Double])])
|
||||
extends Serializable
|
||||
{
|
||||
/** Predict the rating of one user for one product. */
|
||||
def predict(user: Int, product: Int): Double = {
|
||||
|
|
Loading…
Reference in a new issue