Commit ed297dc
Convert FTRL code to a core datatable class (#1417)
FTRL code has been converted into a core datatable class:
- added `fit` and `predict` methods for training and making predictions;
- added `reset` method to reset the model;
- added getters and setters for all the model parameters;
- the `model` itself can now be returned and loaded as a separate frame;
- some other refactoring and modifications.
The algo itself was also slightly optimized, however, it still gets the same score on Kaggle.
So we're ok from the accuracy point of view.1 parent 67abbdd commit ed297dc
File tree
9 files changed
+758
-192
lines changed- c
- extras
- datatable/extras
9 files changed
+758
-192
lines changed| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
13 | 13 | | |
14 | 14 | | |
15 | 15 | | |
16 | | - | |
| 16 | + | |
17 | 17 | | |
18 | 18 | | |
19 | 19 | | |
| |||
199 | 199 | | |
200 | 200 | | |
201 | 201 | | |
202 | | - | |
203 | 202 | | |
204 | 203 | | |
205 | 204 | | |
| |||
234 | 233 | | |
235 | 234 | | |
236 | 235 | | |
| 236 | + | |
| 237 | + | |
237 | 238 | | |
238 | 239 | | |
239 | 240 | | |
| |||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
29 | 29 | | |
30 | 30 | | |
31 | 31 | | |
32 | | - | |
33 | 32 | | |
34 | 33 | | |
35 | 34 | | |
| |||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
45 | 45 | | |
46 | 46 | | |
47 | 47 | | |
48 | | - | |
| 48 | + | |
49 | 49 | | |
50 | | - | |
| 50 | + | |
51 | 51 | | |
52 | 52 | | |
53 | 53 | | |
| |||
0 commit comments