Skip to content

Instantly share code, notes, and snippets.

@gabraganca

gabraganca/svd_training.py

Created April 5, 2019 13:26
Show Gist options
  • Save gabraganca/9744d4b6d2789bd1e2d6209d3f82dd35 to your computer and use it in GitHub Desktop.
Save gabraganca/9744d4b6d2789bd1e2d6209d3f82dd35 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
svd_training.py
# Treinando algoritmo obtido com menor MAE
algo_svd_mae = gs.best_estimator['mae']
algo_svd_mae.fit(full_trainset)
print(get_rec_movies('Esposa', algo_svd_mae)) #Lista de filmes
# Treinando algoritmo obtido com menor RMSE
algo_svd_rmse = gs.best_estimator['rmse']
algo_svd_rmse.fit(full_trainset)
print(get_rec_movies('Esposa', algo_svd_rmse)) #Lista de filmes
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment