From 4c9b5f1a8043e69adde4179442046a5bae3782b9 Mon Sep 17 00:00:00 2001 From: Nil Sagor Date: Wed, 6 Mar 2019 15:13:43 +0800 Subject: [PATCH] "cross_validation" deprecated --- factorizer.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/factorizer.py b/factorizer.py index 0f97029..770a83d 100644 --- a/factorizer.py +++ b/factorizer.py @@ -11,8 +11,8 @@ import pandas as pd import tensorflow as tf import feather -from sklearn.cross_validation import train_test_split -from sklearn.cross_validation import StratifiedKFold +from sklearn.model_selection import train_test_split +from sklearn.model_selection import StratifiedKFold # Given a set of ratings, 2 matrix factors that include one or more # trainable variables, and a regularizer, uses gradient descent to @@ -272,4 +272,4 @@ def main(): np.save("final_h", finalh) if __name__ == '__main__': - main() \ No newline at end of file + main()