libffm 0.1.0

data/vendor/libffm/ffm.h

@@ -0,0 +1,51 @@
+#ifndef _LIBFFM_H
+#define _LIBFFM_H
+
+#include <string>
+
+namespace ffm {
+
+using namespace std;
+
+typedef float ffm_float;
+typedef double ffm_double;
+typedef int ffm_int;
+typedef long long ffm_long;
+
+struct ffm_node {
+    ffm_int f; // field index
+    ffm_int j; // feature index
+    ffm_float v; // value
+};
+
+struct ffm_model {
+    ffm_int n; // number of features
+    ffm_int m; // number of fields
+    ffm_int k; // number of latent factors
+    ffm_float *W = nullptr;
+    bool normalization;
+    void release();
+};
+
+struct ffm_parameter {
+    ffm_float eta = 0.2; // learning rate
+    ffm_float lambda = 0.00002; // regularization parameter
+    ffm_int nr_iters = 15;
+    ffm_int k = 4; // number of latent factors
+    bool normalization = true;
+    bool auto_stop = false;
+};
+
+void ffm_read_problem_to_disk(string txt_path, string bin_path);
+
+void ffm_save_model(ffm_model &model, string path);
+
+ffm_model ffm_load_model(string path);
+
+ffm_model ffm_train_on_disk(string Tr_path, string Va_path, ffm_parameter param);
+
+ffm_float ffm_predict(ffm_node *begin, ffm_node *end, ffm_model &model);
+
+} // namespace ffm
+
+#endif // _LIBFFM_H

data/vendor/libffm/timer.cpp

@@ -0,0 +1,31 @@
+#include <string>
+#include "timer.h"
+
+Timer::Timer()
+{
+    reset();
+}
+
+void Timer::reset()
+{
+    begin = std::chrono::high_resolution_clock::now();
+    duration = 
+        std::chrono::duration_cast<std::chrono::milliseconds>(begin-begin);
+}
+
+void Timer::tic()
+{
+    begin = std::chrono::high_resolution_clock::now();
+}
+
+float Timer::toc()
+{
+    duration += std::chrono::duration_cast<std::chrono::milliseconds>
+                    (std::chrono::high_resolution_clock::now()-begin);
+    return get();
+}
+
+float Timer::get()
+{
+    return (float)duration.count() / 1000;
+}

data/vendor/libffm/timer.h

@@ -0,0 +1,14 @@
+#include <chrono>
+
+class Timer
+{
+public:
+    Timer();
+    void reset();
+    void tic();
+    float toc();
+    float get();
+private:
+    std::chrono::high_resolution_clock::time_point begin;
+    std::chrono::milliseconds duration;
+};

metadata

@@ -0,0 +1,74 @@
+--- !ruby/object:Gem::Specification
+name: libffm
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Andrew Kane
+autorequire:
+bindir: bin
+cert_chain: []
+date: 2020-11-29 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: rice
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '2.2'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '2.2'
+description:
+email: andrew@chartkick.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- CHANGELOG.md
+- LICENSE.txt
+- README.md
+- ext/libffm/ext.cpp
+- ext/libffm/extconf.rb
+- lib/libffm.rb
+- lib/libffm/model.rb
+- lib/libffm/version.rb
+- vendor/libffm/COPYRIGHT
+- vendor/libffm/Makefile
+- vendor/libffm/Makefile.win
+- vendor/libffm/README
+- vendor/libffm/ffm-predict.cpp
+- vendor/libffm/ffm-train.cpp
+- vendor/libffm/ffm.cpp
+- vendor/libffm/ffm.h
+- vendor/libffm/timer.cpp
+- vendor/libffm/timer.h
+homepage: https://github.com/ankane/libffm
+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.5'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.1.4
+signing_key:
+specification_version: 4
+summary: Field-aware factorization machines for Ruby
+test_files: []