libmf 0.1.0

data/vendor/libmf/mf.def

@@ -0,0 +1,21 @@
+LIBRARY mf
+EXPORTS
+    mf_get_default_param    @1
+    read_problem    @2
+    mf_save_model    @3
+    mf_load_model    @4
+    mf_destroy_model    @5
+    mf_train    @6
+    mf_train_on_disk    @7
+    mf_train_with_validation    @8
+    mf_train_with_validation_on_disk    @9
+    mf_cross_validation    @10
+    mf_cross_validation_on_disk    @11
+    mf_predict    @12
+    calc_rmse    @13
+    calc_mae    @14
+    calc_gkl    @15
+    calc_logloss    @16
+    calc_accuracy    @17
+    calc_mpr    @18
+    calc_auc    @19

data/vendor/libmf/mf.h

@@ -0,0 +1,130 @@
+#ifndef _LIBMF_H
+#define _LIBMF_H
+
+#include <string>
+#include <utility>
+
+#ifdef __cplusplus
+extern "C"
+{
+
+namespace mf
+{
+#endif
+
+typedef float mf_float;
+typedef double mf_double;
+typedef int mf_int;
+typedef long long mf_long;
+
+enum {P_L2_MFR=0, P_L1_MFR=1, P_KL_MFR=2, P_LR_MFC=5, P_L2_MFC=6, P_L1_MFC=7,
+      P_ROW_BPR_MFOC=10, P_COL_BPR_MFOC=11, P_L2_MFOC=12};
+enum {RMSE=0, MAE=1, GKL=2, LOGLOSS=5, ACC=6, ROW_MPR=10, COL_MPR=11,
+      ROW_AUC=12, COL_AUC=13};
+
+struct mf_node
+{
+    mf_int u;
+    mf_int v;
+    mf_float r;
+};
+
+struct mf_problem
+{
+    mf_int m;
+    mf_int n;
+    mf_long nnz;
+    struct mf_node *R;
+};
+
+struct mf_parameter
+{
+    mf_int fun;
+    mf_int k;
+    mf_int nr_threads;
+    mf_int nr_bins;
+    mf_int nr_iters;
+    mf_float lambda_p1;
+    mf_float lambda_p2;
+    mf_float lambda_q1;
+    mf_float lambda_q2;
+    mf_float eta;
+    mf_float alpha;
+    mf_float c;
+    bool do_nmf;
+    bool quiet;
+    bool copy_data;
+};
+
+struct mf_parameter mf_get_default_param();
+
+struct mf_model
+{
+    mf_int fun;
+    mf_int m;
+    mf_int n;
+    mf_int k;
+    mf_float b;
+    mf_float *P;
+    mf_float *Q;
+};
+
+mf_problem read_problem(std::string path);
+
+mf_int mf_save_model(struct mf_model const *model, char const *path);
+
+struct mf_model* mf_load_model(char const *path);
+
+void mf_destroy_model(struct mf_model **model);
+
+struct mf_model* mf_train(
+    struct mf_problem const *prob,
+    struct mf_parameter param);
+
+struct mf_model* mf_train_on_disk(
+    char const *tr_path,
+    struct mf_parameter param);
+
+struct mf_model* mf_train_with_validation(
+    struct mf_problem const *tr,
+    struct mf_problem const *va,
+    struct mf_parameter param);
+
+struct mf_model* mf_train_with_validation_on_disk(
+    char const *tr_path,
+    char const *va_path,
+    struct mf_parameter param);
+
+mf_double mf_cross_validation(
+    struct mf_problem const *prob,
+    mf_int nr_folds,
+    struct mf_parameter param);
+
+mf_double mf_cross_validation_on_disk(
+    char const *prob,
+    mf_int nr_folds,
+    mf_parameter param);
+
+mf_float mf_predict(struct mf_model const *model, mf_int u, mf_int v);
+
+mf_double calc_rmse(mf_problem *prob, mf_model *model);
+
+mf_double calc_mae(mf_problem *prob, mf_model *model);
+
+mf_double calc_gkl(mf_problem *prob, mf_model *model);
+
+mf_double calc_logloss(mf_problem *prob, mf_model *model);
+
+mf_double calc_accuracy(mf_problem *prob, mf_model *model);
+
+mf_double calc_mpr(mf_problem *prob, mf_model *model, bool transpose);
+
+mf_double calc_auc(mf_problem *prob, mf_model *model, bool transpose);
+
+#ifdef __cplusplus
+} // namespace mf
+
+} // extern "C"
+#endif
+
+#endif // _LIBMF_H

metadata

@@ -0,0 +1,142 @@
+--- !ruby/object:Gem::Specification
+name: libmf
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Andrew Kane
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2019-11-06 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: ffi
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: minitest
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '5'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '5'
+- !ruby/object:Gem::Dependency
+  name: rake-compiler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+description: 
+email: andrew@chartkick.com
+executables: []
+extensions:
+- ext/libmf/extconf.rb
+extra_rdoc_files: []
+files:
+- CHANGELOG.md
+- LICENSE.txt
+- README.md
+- ext/libmf/extconf.rb
+- lib/libmf.bundle
+- lib/libmf.rb
+- lib/libmf/ffi.rb
+- lib/libmf/model.rb
+- lib/libmf/version.rb
+- vendor/libmf/COPYRIGHT
+- vendor/libmf/Makefile
+- vendor/libmf/Makefile.win
+- vendor/libmf/README
+- vendor/libmf/demo/all_one_matrix.te.txt
+- vendor/libmf/demo/all_one_matrix.tr.txt
+- vendor/libmf/demo/binary_matrix.te.txt
+- vendor/libmf/demo/binary_matrix.tr.txt
+- vendor/libmf/demo/demo.bat
+- vendor/libmf/demo/demo.sh
+- vendor/libmf/demo/real_matrix.te.txt
+- vendor/libmf/demo/real_matrix.tr.txt
+- vendor/libmf/mf-predict.cpp
+- vendor/libmf/mf-train.cpp
+- vendor/libmf/mf.cpp
+- vendor/libmf/mf.def
+- vendor/libmf/mf.h
+- vendor/libmf/windows/mf-predict.exe
+- vendor/libmf/windows/mf-train.exe
+- vendor/libmf/windows/mf.dll
+homepage: https://github.com/ankane/libmf
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '2.4'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.0.3
+signing_key: 
+specification_version: 4
+summary: LIBMF - large-scale sparse matrix factorization - for Ruby
+test_files: []