spark-libFM
spark-libFM copied to clipboard
An implement of Factorization Machines (LibFM)
`task` is currently an int: ``` @param task 0 for Regression, and 1 for Binary Classification ``` Calling it looks like: ``` FMWithSGD.train(trainingData.rdd(), 1, numIterations); ``` Which is much less...
`saveAsParquetFile` should be replaced by `write.parquet` And `parquetFile` should be replaced by `read.parquet`
In FMWithSGD: numFeaturs -> numFeatures FMModle -> FMModel
None of the dependencies specified in the build file are actually used except for spark-mllib. It would be good to delete the others
Hi Ruifeng, I have run a dataset thru your FM engine, which I generated, and contains specific cross term relationships. With Rendle’s libfm project I am getting the expected results,...