Skip to content

Instantly share code, notes, and snippets.

@grayskripko
Last active July 31, 2017 21:36
Show Gist options
  • Save grayskripko/ff1b17e897dfba51895c117bd0b0cc96 to your computer and use it in GitHub Desktop.
Save grayskripko/ff1b17e897dfba51895c117bd0b0cc96 to your computer and use it in GitHub Desktop.
Flawed models
>> Regression:
dataset: [diamonds], target: [price], by_col: [color], prod(dim(df)): [539400]
model time memory mse R2
1: svmRadial 80.8 16.6 MB 403177.8 0.974
2: xgb 1.0 4.1 MB 1201304.2 0.924
3: glm 0.5 19.9 MB 1585278.7 0.899
4: mean_dummy 0.0 0 15767646.5 0.000
....
model time memory mse R2
1: xgb 1.5 4.15 MB 1139890 0.927
2: glm 0.7 17.7 MB 1447223 0.907
3: svmLinear 611.9 9.82 MB 1598951 0.897
4: mean_dummy 0.0 0 15576788 0.000
....
model time memory mse R2
1: rf 1183.7 451 MB 293562.4 0.982
2: ranger 3.2 180 MB 295434.8 0.982
3: catboost 1.8 4.06 MB 471722.2 0.970
4: kknn 40.1 16.3 MB 563559.2 0.965
5: xgb 0.8 3.76 MB 1172524.7 0.927
6: lgbm 0.4 3.64 MB 1405104.6 0.912
7: glm 0.6 17.7 MB 1528333.1 0.904
8: glmnet 1.0 8.48 MB 1573850.5 0.902
9: glm_by 2.1 19.7 MB 1610757.8 0.899
10: mean_dummy 0.0 0 15985243.9 0.000
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment