Skip to content

Instantly share code, notes, and snippets.

@coreylynch
Created December 1, 2015 15:20
Show Gist options
  • Save coreylynch/9ed4f43ab5db53f38499 to your computer and use it in GitHub Desktop.
Save coreylynch/9ed4f43ab5db53f38499 to your computer and use it in GitHub Desktop.
In [19]: fm = pylibfm.FM(num_factors=10, num_iter=300, verbose=True, task="regression", initial_learning_rate=0.0001, learning_rate_schedule=
"optimal")
In [20]: fm.fit(X_train,y_train)
Creating validation dataset of 0.01 of training for adaptive regularization
-- Epoch 1
Training RMSE: 0.78393
-- Epoch 2
Training RMSE: 0.61129
-- Epoch 3
Training RMSE: 0.59825
-- Epoch 4
Training RMSE: 0.58678
-- Epoch 5
Training RMSE: 0.57664
-- Epoch 6
Training RMSE: 0.56759
-- Epoch 7
Training RMSE: 0.55950
-- Epoch 8
Training RMSE: 0.55225
-- Epoch 9
Training RMSE: 0.54564
-- Epoch 10
Training RMSE: 0.53966
-- Epoch 11
Training RMSE: 0.53419
-- Epoch 12
Training RMSE: 0.52916
-- Epoch 13
Training RMSE: 0.52457
-- Epoch 14
Training RMSE: 0.52032
-- Epoch 15
Training RMSE: 0.51637
-- Epoch 16
Training RMSE: 0.51271
-- Epoch 17
Training RMSE: 0.50930
-- Epoch 18
Training RMSE: 0.50611
-- Epoch 19
Training RMSE: 0.50316
-- Epoch 20
Training RMSE: 0.50035
-- Epoch 21
Training RMSE: 0.49773
-- Epoch 22
Training RMSE: 0.49527
-- Epoch 23
Training RMSE: 0.49293
-- Epoch 24
Training RMSE: 0.49074
-- Epoch 25
Training RMSE: 0.48868
-- Epoch 26
Training RMSE: 0.48669
-- Epoch 27
Training RMSE: 0.48484
-- Epoch 28
Training RMSE: 0.48303
-- Epoch 29
Training RMSE: 0.48137
-- Epoch 30
Training RMSE: 0.47974
-- Epoch 31
Training RMSE: 0.47824
-- Epoch 32
Training RMSE: 0.47675
-- Epoch 33
Training RMSE: 0.47533
-- Epoch 34
Training RMSE: 0.47400
-- Epoch 35
Training RMSE: 0.47271
-- Epoch 36
Training RMSE: 0.47145
-- Epoch 37
Training RMSE: 0.47028
-- Epoch 38
Training RMSE: 0.46916
-- Epoch 39
Training RMSE: 0.46804
-- Epoch 40
Training RMSE: 0.46699
-- Epoch 41
Training RMSE: 0.46594
-- Epoch 42
Training RMSE: 0.46497
-- Epoch 43
Training RMSE: 0.46399
-- Epoch 44
Training RMSE: 0.46310
-- Epoch 45
Training RMSE: 0.46221
-- Epoch 46
Training RMSE: 0.46136
-- Epoch 47
Training RMSE: 0.46052
-- Epoch 48
Training RMSE: 0.45970
-- Epoch 49
Training RMSE: 0.45893
-- Epoch 50
Training RMSE: 0.45818
-- Epoch 51
Training RMSE: 0.45742
-- Epoch 52
Training RMSE: 0.45670
-- Epoch 53
Training RMSE: 0.45601
-- Epoch 54
Training RMSE: 0.45530
-- Epoch 55
Training RMSE: 0.45468
-- Epoch 56
Training RMSE: 0.45404
-- Epoch 57
Training RMSE: 0.45340
-- Epoch 58
Training RMSE: 0.45280
-- Epoch 59
Training RMSE: 0.45218
-- Epoch 60
Training RMSE: 0.45164
-- Epoch 61
Training RMSE: 0.45106
-- Epoch 62
Training RMSE: 0.45052
-- Epoch 63
Training RMSE: 0.44998
-- Epoch 64
Training RMSE: 0.44945
-- Epoch 65
Training RMSE: 0.44895
-- Epoch 66
Training RMSE: 0.44843
-- Epoch 67
Training RMSE: 0.44795
-- Epoch 68
Training RMSE: 0.44746
-- Epoch 69
Training RMSE: 0.44701
-- Epoch 70
Training RMSE: 0.44652
-- Epoch 71
Training RMSE: 0.44606
-- Epoch 72
Training RMSE: 0.44565
-- Epoch 73
Training RMSE: 0.44519
-- Epoch 74
Training RMSE: 0.44478
-- Epoch 75
Training RMSE: 0.44436
-- Epoch 76
Training RMSE: 0.44393
-- Epoch 77
Training RMSE: 0.44355
-- Epoch 78
Training RMSE: 0.44317
-- Epoch 79
Training RMSE: 0.44279
-- Epoch 80
Training RMSE: 0.44240
-- Epoch 81
Training RMSE: 0.44201
-- Epoch 82
Training RMSE: 0.44165
-- Epoch 83
Training RMSE: 0.44131
-- Epoch 84
Training RMSE: 0.44095
-- Epoch 85
Training RMSE: 0.44058
-- Epoch 86
Training RMSE: 0.44027
-- Epoch 87
Training RMSE: 0.43991
-- Epoch 88
Training RMSE: 0.43958
-- Epoch 89
Training RMSE: 0.43925
-- Epoch 90
Training RMSE: 0.43891
-- Epoch 91
Training RMSE: 0.43862
-- Epoch 92
Training RMSE: 0.43831
-- Epoch 93
Training RMSE: 0.43801
-- Epoch 94
Training RMSE: 0.43770
-- Epoch 95
Training RMSE: 0.43743
-- Epoch 96
Training RMSE: 0.43712
-- Epoch 97
Training RMSE: 0.43684
-- Epoch 98
Training RMSE: 0.43655
-- Epoch 99
Training RMSE: 0.43627
-- Epoch 100
Training RMSE: 0.43598
-- Epoch 101
Training RMSE: 0.43572
-- Epoch 102
Training RMSE: 0.43545
-- Epoch 103
Training RMSE: 0.43519
-- Epoch 104
Training RMSE: 0.43494
-- Epoch 105
Training RMSE: 0.43466
-- Epoch 106
Training RMSE: 0.43440
-- Epoch 107
Training RMSE: 0.43416
-- Epoch 108
Training RMSE: 0.43391
-- Epoch 109
Training RMSE: 0.43364
-- Epoch 110
Training RMSE: 0.43342
-- Epoch 111
Training RMSE: 0.43318
-- Epoch 112
Training RMSE: 0.43294
-- Epoch 113
Training RMSE: 0.43271
-- Epoch 114
Training RMSE: 0.43249
-- Epoch 115
Training RMSE: 0.43225
-- Epoch 116
Training RMSE: 0.43204
-- Epoch 117
Training RMSE: 0.43179
-- Epoch 118
Training RMSE: 0.43160
-- Epoch 119
Training RMSE: 0.43138
-- Epoch 120
Training RMSE: 0.43115
-- Epoch 121
Training RMSE: 0.43094
-- Epoch 122
Training RMSE: 0.43073
-- Epoch 123
Training RMSE: 0.43049
-- Epoch 124
Training RMSE: 0.43032
-- Epoch 125
Training RMSE: 0.43008
-- Epoch 126
Training RMSE: 0.42990
-- Epoch 127
Training RMSE: 0.42971
-- Epoch 128
Training RMSE: 0.42949
-- Epoch 129
Training RMSE: 0.42930
-- Epoch 130
Training RMSE: 0.42912
-- Epoch 131
Training RMSE: 0.42891
-- Epoch 132
Training RMSE: 0.42873
-- Epoch 133
Training RMSE: 0.42854
-- Epoch 134
Training RMSE: 0.42834
-- Epoch 135
Training RMSE: 0.42816
-- Epoch 136
Training RMSE: 0.42800
-- Epoch 137
Training RMSE: 0.42781
-- Epoch 138
Training RMSE: 0.42762
-- Epoch 139
Training RMSE: 0.42743
-- Epoch 140
Training RMSE: 0.42725
-- Epoch 141
Training RMSE: 0.42707
-- Epoch 142
Training RMSE: 0.42691
-- Epoch 143
Training RMSE: 0.42672
-- Epoch 144
Training RMSE: 0.42658
-- Epoch 145
Training RMSE: 0.42639
-- Epoch 146
Training RMSE: 0.42621
-- Epoch 147
Training RMSE: 0.42602
-- Epoch 148
Training RMSE: 0.42589
-- Epoch 149
Training RMSE: 0.42574
-- Epoch 150
Training RMSE: 0.42555
-- Epoch 151
Training RMSE: 0.42537
-- Epoch 152
Training RMSE: 0.42521
-- Epoch 153
Training RMSE: 0.42507
-- Epoch 154
Training RMSE: 0.42492
-- Epoch 155
Training RMSE: 0.42476
-- Epoch 156
Training RMSE: 0.42459
-- Epoch 157
Training RMSE: 0.42442
-- Epoch 158
Training RMSE: 0.42428
-- Epoch 159
Training RMSE: 0.42413
-- Epoch 160
Training RMSE: 0.42396
-- Epoch 161
Training RMSE: 0.42384
-- Epoch 162
Training RMSE: 0.42365
-- Epoch 163
Training RMSE: 0.42353
-- Epoch 164
Training RMSE: 0.42336
-- Epoch 165
Training RMSE: 0.42319
-- Epoch 166
Training RMSE: 0.42308
-- Epoch 167
Training RMSE: 0.42293
-- Epoch 168
Training RMSE: 0.42278
-- Epoch 169
Training RMSE: 0.42265
-- Epoch 170
Training RMSE: 0.42249
-- Epoch 171
Training RMSE: 0.42231
-- Epoch 172
Training RMSE: 0.42219
-- Epoch 173
Training RMSE: 0.42204
-- Epoch 174
Training RMSE: 0.42190
-- Epoch 175
Training RMSE: 0.42179
-- Epoch 176
Training RMSE: 0.42162
-- Epoch 177
Training RMSE: 0.42150
-- Epoch 178
Training RMSE: 0.42135
-- Epoch 179
Training RMSE: 0.42121
-- Epoch 180
Training RMSE: 0.42109
-- Epoch 181
Training RMSE: 0.42093
-- Epoch 182
Training RMSE: 0.42081
-- Epoch 183
Training RMSE: 0.42066
-- Epoch 184
Training RMSE: 0.42053
-- Epoch 185
Training RMSE: 0.42039
-- Epoch 186
Training RMSE: 0.42026
-- Epoch 187
Training RMSE: 0.42009
-- Epoch 188
Training RMSE: 0.41998
-- Epoch 189
Training RMSE: 0.41984
-- Epoch 190
Training RMSE: 0.41968
-- Epoch 191
Training RMSE: 0.41961
-- Epoch 192
Training RMSE: 0.41946
-- Epoch 193
Training RMSE: 0.41932
-- Epoch 194
Training RMSE: 0.41920
-- Epoch 195
Training RMSE: 0.41906
-- Epoch 196
Training RMSE: 0.41892
-- Epoch 197
Training RMSE: 0.41880
-- Epoch 198
Training RMSE: 0.41869
-- Epoch 199
Training RMSE: 0.41855
-- Epoch 200
Training RMSE: 0.41842
-- Epoch 201
Training RMSE: 0.41827
-- Epoch 202
Training RMSE: 0.41816
-- Epoch 203
Training RMSE: 0.41801
-- Epoch 204
Training RMSE: 0.41791
-- Epoch 205
Training RMSE: 0.41774
-- Epoch 206
Training RMSE: 0.41766
-- Epoch 207
Training RMSE: 0.41751
-- Epoch 208
Training RMSE: 0.41742
-- Epoch 209
Training RMSE: 0.41727
-- Epoch 210
Training RMSE: 0.41715
-- Epoch 211
Training RMSE: 0.41700
-- Epoch 212
Training RMSE: 0.41689
-- Epoch 213
Training RMSE: 0.41675
-- Epoch 214
Training RMSE: 0.41665
-- Epoch 215
Training RMSE: 0.41648
-- Epoch 216
Training RMSE: 0.41641
-- Epoch 217
Training RMSE: 0.41628
-- Epoch 218
Training RMSE: 0.41615
-- Epoch 219
Training RMSE: 0.41601
-- Epoch 220
Training RMSE: 0.41589
-- Epoch 221
Training RMSE: 0.41579
-- Epoch 222
Training RMSE: 0.41567
-- Epoch 223
Training RMSE: 0.41550
-- Epoch 224
Training RMSE: 0.41541
-- Epoch 225
Training RMSE: 0.41525
-- Epoch 226
Training RMSE: 0.41516
-- Epoch 227
Training RMSE: 0.41504
-- Epoch 228
Training RMSE: 0.41485
-- Epoch 229
Training RMSE: 0.41479
-- Epoch 230
Training RMSE: 0.41466
-- Epoch 231
Training RMSE: 0.41452
-- Epoch 232
Training RMSE: 0.41442
-- Epoch 233
Training RMSE: 0.41430
-- Epoch 234
Training RMSE: 0.41415
-- Epoch 235
Training RMSE: 0.41404
-- Epoch 236
Training RMSE: 0.41392
-- Epoch 237
Training RMSE: 0.41381
-- Epoch 238
Training RMSE: 0.41366
-- Epoch 239
Training RMSE: 0.41354
-- Epoch 240
Training RMSE: 0.41343
-- Epoch 241
Training RMSE: 0.41329
-- Epoch 242
Training RMSE: 0.41319
-- Epoch 243
Training RMSE: 0.41306
-- Epoch 244
Training RMSE: 0.41292
-- Epoch 245
Training RMSE: 0.41276
-- Epoch 246
Training RMSE: 0.41266
-- Epoch 247
Training RMSE: 0.41257
-- Epoch 248
Training RMSE: 0.41244
-- Epoch 249
Training RMSE: 0.41230
-- Epoch 250
Training RMSE: 0.41217
-- Epoch 251
Training RMSE: 0.41207
-- Epoch 252
Training RMSE: 0.41195
-- Epoch 253
Training RMSE: 0.41182
-- Epoch 254
Training RMSE: 0.41167
-- Epoch 255
Training RMSE: 0.41156
-- Epoch 256
Training RMSE: 0.41143
-- Epoch 257
Training RMSE: 0.41131
-- Epoch 258
Training RMSE: 0.41119
-- Epoch 259
Training RMSE: 0.41107
-- Epoch 260
Training RMSE: 0.41093
-- Epoch 261
Training RMSE: 0.41080
-- Epoch 262
Training RMSE: 0.41067
-- Epoch 263
Training RMSE: 0.41056
-- Epoch 264
Training RMSE: 0.41040
-- Epoch 265
Training RMSE: 0.41031
-- Epoch 266
Training RMSE: 0.41017
-- Epoch 267
Training RMSE: 0.41004
-- Epoch 268
Training RMSE: 0.40992
-- Epoch 269
Training RMSE: 0.40981
-- Epoch 270
Training RMSE: 0.40967
-- Epoch 271
Training RMSE: 0.40954
-- Epoch 272
Training RMSE: 0.40941
-- Epoch 273
Training RMSE: 0.40929
-- Epoch 274
Training RMSE: 0.40916
-- Epoch 275
Training RMSE: 0.40901
-- Epoch 276
Training RMSE: 0.40888
-- Epoch 277
Training RMSE: 0.40875
-- Epoch 278
Training RMSE: 0.40862
-- Epoch 279
Training RMSE: 0.40848
-- Epoch 280
Training RMSE: 0.40837
-- Epoch 281
Training RMSE: 0.40822
-- Epoch 282
Training RMSE: 0.40810
-- Epoch 283
Training RMSE: 0.40798
-- Epoch 284
Training RMSE: 0.40784
-- Epoch 285
Training RMSE: 0.40771
-- Epoch 286
Training RMSE: 0.40754
-- Epoch 287
Training RMSE: 0.40746
-- Epoch 288
Training RMSE: 0.40731
-- Epoch 289
Training RMSE: 0.40717
-- Epoch 290
Training RMSE: 0.40703
-- Epoch 291
Training RMSE: 0.40690
-- Epoch 292
Training RMSE: 0.40676
-- Epoch 293
Training RMSE: 0.40665
-- Epoch 294
Training RMSE: 0.40651
-- Epoch 295
Training RMSE: 0.40638
-- Epoch 296
Training RMSE: 0.40623
-- Epoch 297
Training RMSE: 0.40609
-- Epoch 298
Training RMSE: 0.40596
-- Epoch 299
Training RMSE: 0.40579
-- Epoch 300
Training RMSE: 0.40568
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment