├── LICENSE ├── README.md ├── data ├── a9a.te └── a9a.tr ├── project ├── build.properties └── plugins.sbt └── src ├── main └── scala │ └── org │ └── apache │ └── spark │ └── ml │ ├── fm │ └── FactorizationMachines.scala │ └── optim │ ├── ParallelFtrl.scala │ ├── ParallelGradientDescent.scala │ ├── PerCoordinateUpdater.scala │ └── configuration │ ├── Algo.scala │ └── Solver.scala └── test └── scala └── org └── apache └── spark └── ml └── fm └── FactorizationMachinesSuite.scala /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hibayesian/spark-fm/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hibayesian/spark-fm/HEAD/README.md -------------------------------------------------------------------------------- /data/a9a.te: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hibayesian/spark-fm/HEAD/data/a9a.te -------------------------------------------------------------------------------- /data/a9a.tr: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hibayesian/spark-fm/HEAD/data/a9a.tr -------------------------------------------------------------------------------- /project/build.properties: -------------------------------------------------------------------------------- 1 | sbt.version = 0.13.6 2 | -------------------------------------------------------------------------------- /project/plugins.sbt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hibayesian/spark-fm/HEAD/project/plugins.sbt -------------------------------------------------------------------------------- /src/main/scala/org/apache/spark/ml/fm/FactorizationMachines.scala: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hibayesian/spark-fm/HEAD/src/main/scala/org/apache/spark/ml/fm/FactorizationMachines.scala -------------------------------------------------------------------------------- /src/main/scala/org/apache/spark/ml/optim/ParallelFtrl.scala: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hibayesian/spark-fm/HEAD/src/main/scala/org/apache/spark/ml/optim/ParallelFtrl.scala -------------------------------------------------------------------------------- /src/main/scala/org/apache/spark/ml/optim/ParallelGradientDescent.scala: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hibayesian/spark-fm/HEAD/src/main/scala/org/apache/spark/ml/optim/ParallelGradientDescent.scala -------------------------------------------------------------------------------- /src/main/scala/org/apache/spark/ml/optim/PerCoordinateUpdater.scala: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hibayesian/spark-fm/HEAD/src/main/scala/org/apache/spark/ml/optim/PerCoordinateUpdater.scala -------------------------------------------------------------------------------- /src/main/scala/org/apache/spark/ml/optim/configuration/Algo.scala: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hibayesian/spark-fm/HEAD/src/main/scala/org/apache/spark/ml/optim/configuration/Algo.scala -------------------------------------------------------------------------------- /src/main/scala/org/apache/spark/ml/optim/configuration/Solver.scala: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hibayesian/spark-fm/HEAD/src/main/scala/org/apache/spark/ml/optim/configuration/Solver.scala -------------------------------------------------------------------------------- /src/test/scala/org/apache/spark/ml/fm/FactorizationMachinesSuite.scala: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hibayesian/spark-fm/HEAD/src/test/scala/org/apache/spark/ml/fm/FactorizationMachinesSuite.scala --------------------------------------------------------------------------------