xlearn 0.1.2 → 0.1.3
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- checksums.yaml +4 -4
- data/CHANGELOG.md +4 -0
- data/README.md +23 -6
- data/lib/xlearn/version.rb +1 -1
- data/vendor/libxlearn_api.dylib +0 -0
- data/vendor/libxlearn_api.so +0 -0
- metadata +2 -2
checksums.yaml
CHANGED
@@ -1,7 +1,7 @@
|
|
1
1
|
---
|
2
2
|
SHA256:
|
3
|
-
metadata.gz:
|
4
|
-
data.tar.gz:
|
3
|
+
metadata.gz: e412507862a4f703ac75cfb7c69a8f60bafdcafc20650e985607929c7d2a1682
|
4
|
+
data.tar.gz: c750ba69fcb27cb436d597cc6f5825efd2ca9a818cdf8ef55343f509939e0f0e
|
5
5
|
SHA512:
|
6
|
-
metadata.gz:
|
7
|
-
data.tar.gz:
|
6
|
+
metadata.gz: 3ad86fc29459c7f6a8a9bfce72ee3bba772f39b6279804d4c758d1f93786e0dff4a0db04b31d75b1fe7afb9f110eeabab11ba4c0a68df9138f65ee2db60bc225
|
7
|
+
data.tar.gz: d5c862cf8b0516c2c29436c9f2c735f1bf2f20b68aae9cff40bfcc2d71f10df866438ed46ae25d623c87ec3bcbe22648b0c92784435126c0f01d579119e702e5
|
data/CHANGELOG.md
CHANGED
data/README.md
CHANGED
@@ -82,14 +82,31 @@ model.latent_factors # fm and ffm only
|
|
82
82
|
|
83
83
|
## Parameters
|
84
84
|
|
85
|
-
|
86
|
-
|
87
|
-
```ruby
|
88
|
-
|
85
|
+
Pass parameters - default values below
|
86
|
+
|
87
|
+
```ruby
|
88
|
+
XLearn::FM.new(
|
89
|
+
task: "binary", # binary (classification), reg (regression)
|
90
|
+
metric: nil, # acc, prec, recall, f1, auc, mae, mape, rmse, rmsd
|
91
|
+
lr: 0.2, # learning rate
|
92
|
+
lambda: 0.00002, # lambda for l2 regularization
|
93
|
+
k: 4, # latent factors for fm and ffm
|
94
|
+
alpha: 0.3, # hyper parameter for ftrl
|
95
|
+
beta: 1.0, # hyper parameter for ftrl
|
96
|
+
lambda_1: 0.00001, # hyper parameter for ftrl
|
97
|
+
lambda_2: 0.00002, # hyper parameter for ftrl
|
98
|
+
epoch: 10, # number of epochs
|
99
|
+
fold: 3, # number of folds
|
100
|
+
opt: "adagrad", # sgd, adagrad, ftrl
|
101
|
+
block_size: 500, # block size for on-disk training in MB
|
102
|
+
early_stop: true, # use early stopping
|
103
|
+
stop_window: 2, # size of stop window for early stopping
|
104
|
+
sign: false, # convert predition output to 0 and 1
|
105
|
+
sigmoid: false, # convert predition output using sigmoid
|
106
|
+
seed: 1 # random seed to shuffle data set
|
107
|
+
)
|
89
108
|
```
|
90
109
|
|
91
|
-
Supports the same parameters as [Python](https://xlearn-doc.readthedocs.io/en/latest/all_api/index.html)
|
92
|
-
|
93
110
|
## Cross-Validation
|
94
111
|
|
95
112
|
Cross-validation
|
data/lib/xlearn/version.rb
CHANGED
data/vendor/libxlearn_api.dylib
CHANGED
Binary file
|
data/vendor/libxlearn_api.so
CHANGED
Binary file
|
metadata
CHANGED
@@ -1,14 +1,14 @@
|
|
1
1
|
--- !ruby/object:Gem::Specification
|
2
2
|
name: xlearn
|
3
3
|
version: !ruby/object:Gem::Version
|
4
|
-
version: 0.1.
|
4
|
+
version: 0.1.3
|
5
5
|
platform: ruby
|
6
6
|
authors:
|
7
7
|
- Andrew Kane
|
8
8
|
autorequire:
|
9
9
|
bindir: bin
|
10
10
|
cert_chain: []
|
11
|
-
date: 2020-
|
11
|
+
date: 2020-02-10 00:00:00.000000000 Z
|
12
12
|
dependencies:
|
13
13
|
- !ruby/object:Gem::Dependency
|
14
14
|
name: ffi
|