├── .gitignore ├── LICENSE ├── Makefile ├── README.md ├── docs ├── orignal_ffm_y.png ├── simplified_ffm_y.png ├── w_jf.png ├── w_size.png ├── w_xf_size.png └── x_j.png ├── examples ├── dataset1.txt └── dataset2.txt ├── java ├── pom.xml └── src │ ├── main │ ├── java │ │ └── com │ │ │ └── rtbhouse │ │ │ └── model │ │ │ └── natives │ │ │ └── FactorizationMachineNativeOps.java │ └── resources │ │ └── com │ │ └── rtbhouse │ │ └── model │ │ └── natives │ │ └── FactorizationMachineNativeOps.h │ └── test │ └── java │ └── com │ └── rtbhouse │ └── model │ └── natives │ └── FactorizationMachinesNativeOpsTest.java ├── src ├── benchmark.cpp ├── bin_to_text.cpp ├── cli.h ├── constants.h ├── cuda_buffer.h ├── cuda_utils.cu ├── cuda_utils.h ├── dataset.h ├── dataset_writer.h ├── ffm_predictor.cu ├── ffm_predictor.h ├── ffm_trainer.cu ├── ffm_trainer.h ├── log_loss_calculator.h ├── model.cpp ├── model.h ├── predict.cpp ├── shuffler.cpp ├── splitter.cpp ├── text_to_bin.cpp ├── timer.h ├── trainer.cpp ├── training_history.cpp ├── training_history.h ├── training_options.cpp ├── training_options.h ├── training_session.cpp ├── training_session.h └── utils.h └── tests └── tests.sh /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RTBHOUSE/cuda-ffm/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RTBHOUSE/cuda-ffm/HEAD/LICENSE -------------------------------------------------------------------------------- /Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RTBHOUSE/cuda-ffm/HEAD/Makefile -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RTBHOUSE/cuda-ffm/HEAD/README.md -------------------------------------------------------------------------------- /docs/orignal_ffm_y.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RTBHOUSE/cuda-ffm/HEAD/docs/orignal_ffm_y.png -------------------------------------------------------------------------------- /docs/simplified_ffm_y.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RTBHOUSE/cuda-ffm/HEAD/docs/simplified_ffm_y.png -------------------------------------------------------------------------------- /docs/w_jf.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RTBHOUSE/cuda-ffm/HEAD/docs/w_jf.png -------------------------------------------------------------------------------- /docs/w_size.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RTBHOUSE/cuda-ffm/HEAD/docs/w_size.png -------------------------------------------------------------------------------- /docs/w_xf_size.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RTBHOUSE/cuda-ffm/HEAD/docs/w_xf_size.png -------------------------------------------------------------------------------- /docs/x_j.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RTBHOUSE/cuda-ffm/HEAD/docs/x_j.png -------------------------------------------------------------------------------- /examples/dataset1.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RTBHOUSE/cuda-ffm/HEAD/examples/dataset1.txt -------------------------------------------------------------------------------- /examples/dataset2.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RTBHOUSE/cuda-ffm/HEAD/examples/dataset2.txt -------------------------------------------------------------------------------- /java/pom.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RTBHOUSE/cuda-ffm/HEAD/java/pom.xml -------------------------------------------------------------------------------- /java/src/main/java/com/rtbhouse/model/natives/FactorizationMachineNativeOps.java: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RTBHOUSE/cuda-ffm/HEAD/java/src/main/java/com/rtbhouse/model/natives/FactorizationMachineNativeOps.java -------------------------------------------------------------------------------- /java/src/main/resources/com/rtbhouse/model/natives/FactorizationMachineNativeOps.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RTBHOUSE/cuda-ffm/HEAD/java/src/main/resources/com/rtbhouse/model/natives/FactorizationMachineNativeOps.h -------------------------------------------------------------------------------- /java/src/test/java/com/rtbhouse/model/natives/FactorizationMachinesNativeOpsTest.java: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RTBHOUSE/cuda-ffm/HEAD/java/src/test/java/com/rtbhouse/model/natives/FactorizationMachinesNativeOpsTest.java -------------------------------------------------------------------------------- /src/benchmark.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RTBHOUSE/cuda-ffm/HEAD/src/benchmark.cpp -------------------------------------------------------------------------------- /src/bin_to_text.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RTBHOUSE/cuda-ffm/HEAD/src/bin_to_text.cpp -------------------------------------------------------------------------------- /src/cli.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RTBHOUSE/cuda-ffm/HEAD/src/cli.h -------------------------------------------------------------------------------- /src/constants.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RTBHOUSE/cuda-ffm/HEAD/src/constants.h -------------------------------------------------------------------------------- /src/cuda_buffer.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RTBHOUSE/cuda-ffm/HEAD/src/cuda_buffer.h -------------------------------------------------------------------------------- /src/cuda_utils.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RTBHOUSE/cuda-ffm/HEAD/src/cuda_utils.cu -------------------------------------------------------------------------------- /src/cuda_utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RTBHOUSE/cuda-ffm/HEAD/src/cuda_utils.h -------------------------------------------------------------------------------- /src/dataset.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RTBHOUSE/cuda-ffm/HEAD/src/dataset.h -------------------------------------------------------------------------------- /src/dataset_writer.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RTBHOUSE/cuda-ffm/HEAD/src/dataset_writer.h -------------------------------------------------------------------------------- /src/ffm_predictor.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RTBHOUSE/cuda-ffm/HEAD/src/ffm_predictor.cu -------------------------------------------------------------------------------- /src/ffm_predictor.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RTBHOUSE/cuda-ffm/HEAD/src/ffm_predictor.h -------------------------------------------------------------------------------- /src/ffm_trainer.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RTBHOUSE/cuda-ffm/HEAD/src/ffm_trainer.cu -------------------------------------------------------------------------------- /src/ffm_trainer.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RTBHOUSE/cuda-ffm/HEAD/src/ffm_trainer.h -------------------------------------------------------------------------------- /src/log_loss_calculator.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RTBHOUSE/cuda-ffm/HEAD/src/log_loss_calculator.h -------------------------------------------------------------------------------- /src/model.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RTBHOUSE/cuda-ffm/HEAD/src/model.cpp -------------------------------------------------------------------------------- /src/model.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RTBHOUSE/cuda-ffm/HEAD/src/model.h -------------------------------------------------------------------------------- /src/predict.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RTBHOUSE/cuda-ffm/HEAD/src/predict.cpp -------------------------------------------------------------------------------- /src/shuffler.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RTBHOUSE/cuda-ffm/HEAD/src/shuffler.cpp -------------------------------------------------------------------------------- /src/splitter.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RTBHOUSE/cuda-ffm/HEAD/src/splitter.cpp -------------------------------------------------------------------------------- /src/text_to_bin.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RTBHOUSE/cuda-ffm/HEAD/src/text_to_bin.cpp -------------------------------------------------------------------------------- /src/timer.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RTBHOUSE/cuda-ffm/HEAD/src/timer.h -------------------------------------------------------------------------------- /src/trainer.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RTBHOUSE/cuda-ffm/HEAD/src/trainer.cpp -------------------------------------------------------------------------------- /src/training_history.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RTBHOUSE/cuda-ffm/HEAD/src/training_history.cpp -------------------------------------------------------------------------------- /src/training_history.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RTBHOUSE/cuda-ffm/HEAD/src/training_history.h -------------------------------------------------------------------------------- /src/training_options.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RTBHOUSE/cuda-ffm/HEAD/src/training_options.cpp -------------------------------------------------------------------------------- /src/training_options.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RTBHOUSE/cuda-ffm/HEAD/src/training_options.h -------------------------------------------------------------------------------- /src/training_session.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RTBHOUSE/cuda-ffm/HEAD/src/training_session.cpp -------------------------------------------------------------------------------- /src/training_session.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RTBHOUSE/cuda-ffm/HEAD/src/training_session.h -------------------------------------------------------------------------------- /src/utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RTBHOUSE/cuda-ffm/HEAD/src/utils.h -------------------------------------------------------------------------------- /tests/tests.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RTBHOUSE/cuda-ffm/HEAD/tests/tests.sh --------------------------------------------------------------------------------