xlearn 0.1.2 → 0.1.3

Sign up to get free protection for your applications and to get access to all the features.
checksums.yaml CHANGED
@@ -1,7 +1,7 @@
1
1
  ---
2
2
  SHA256:
3
- metadata.gz: 28dfa35c67720ae7fbd0a83c6099cf2e48c8d7bca3d945a53b9a2fdc7b01fdab
4
- data.tar.gz: '0909d3bb3a4353ae0697efd69e109a61cf5d148ef78276f7b7970b87a0eb2e16'
3
+ metadata.gz: e412507862a4f703ac75cfb7c69a8f60bafdcafc20650e985607929c7d2a1682
4
+ data.tar.gz: c750ba69fcb27cb436d597cc6f5825efd2ca9a818cdf8ef55343f509939e0f0e
5
5
  SHA512:
6
- metadata.gz: 06ba4137b93b18a58b31c01140b112e3556f5122913431e731c9a1724a4865c515c96f5183c80e2aacc389c6eecfda91b0f35ed4488946344689ba306e2f94cd
7
- data.tar.gz: 538c6243c42ba5e892a13f13b88f78d57b90bbc27eea4cbd9856944fee723305b301f460de5e55d11d60af5ee260488a190b8a31f548f84868821ebede0b881c
6
+ metadata.gz: 3ad86fc29459c7f6a8a9bfce72ee3bba772f39b6279804d4c758d1f93786e0dff4a0db04b31d75b1fe7afb9f110eeabab11ba4c0a68df9138f65ee2db60bc225
7
+ data.tar.gz: d5c862cf8b0516c2c29436c9f2c735f1bf2f20b68aae9cff40bfcc2d71f10df866438ed46ae25d623c87ec3bcbe22648b0c92784435126c0f01d579119e702e5
data/CHANGELOG.md CHANGED
@@ -1,3 +1,7 @@
1
+ ## 0.1.3 (2020-02-09)
2
+
3
+ - Fixed illegal instruction error on some Linux platforms
4
+
1
5
  ## 0.1.2 (2020-01-26)
2
6
 
3
7
  - Changed to Apache 2.0 license to match xLearn
data/README.md CHANGED
@@ -82,14 +82,31 @@ model.latent_factors # fm and ffm only
82
82
 
83
83
  ## Parameters
84
84
 
85
- Specify parameters
86
-
87
- ```ruby
88
- model = XLearn::Linear.new(k: 20, epoch: 50)
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
@@ -1,3 +1,3 @@
1
1
  module XLearn
2
- VERSION = "0.1.2"
2
+ VERSION = "0.1.3"
3
3
  end
Binary file
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.2
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-01-27 00:00:00.000000000 Z
11
+ date: 2020-02-10 00:00:00.000000000 Z
12
12
  dependencies:
13
13
  - !ruby/object:Gem::Dependency
14
14
  name: ffi