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()