spark-libFM icon indicating copy to clipboard operation
spark-libFM copied to clipboard

An implement of Factorization Machines (LibFM)

Results 15 spark-libFM issues
Sort by recently updated
recently updated
newest added

`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,...