Skip to content
This repository has been archived by the owner on Nov 12, 2024. It is now read-only.

Commit

Permalink
Update to new api
Browse files Browse the repository at this point in the history
  • Loading branch information
gbolmier committed Jan 26, 2021
1 parent 8c2a737 commit 2954007
Showing 1 changed file with 6 additions and 6 deletions.
12 changes: 6 additions & 6 deletions benchmark.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,7 @@
"from sklearn.metrics import mean_squared_error\n",
"\n",
"from funk_svd.dataset import fetch_ml_ratings\n",
"from funk_svd.utils import timer\n",
"from funk_svd.utils import _timer\n",
"from funk_svd import SVD"
]
},
Expand Down Expand Up @@ -52,8 +52,8 @@
"source": [
"%%time\n",
"\n",
"df = fetch_ml_ratings(variant='20m')\n",
"print ()"
"df = fetch_ml_ratings(variant='20m', verbose=True)\n",
"print()"
]
},
{
Expand Down Expand Up @@ -421,10 +421,10 @@
}
],
"source": [
"svd = SVD(learning_rate=0.001, regularization=0.005, n_epochs=100,\n",
" n_factors=15, min_rating=1, max_rating=5)\n",
"svd = SVD(lr=0.001, reg=0.005, n_epochs=100, n_factors=15,\n",
" early_stopping=True, shuffle=False, min_rating=1, max_rating=5)\n",
"\n",
"svd.fit(X=train, X_val=val, early_stopping=True, shuffle=False)"
"svd.fit(X=train, X_val=val)"
]
},
{
Expand Down

0 comments on commit 2954007

Please sign in to comment.