diff --git a/build.py b/build.py index dc83f4ba41a35f0c4eec562e99bf64b1ebef57f5..3e7c174544f63811e2f7aadbec0eaa3668ec50df 100644 --- a/build.py +++ b/build.py @@ -1,4 +1,6 @@ #Imports +import warnings +warnings.filterwarnings('ignore') import re from random import seed import pandas as pd @@ -18,9 +20,6 @@ from sklearn.feature_extraction.text import CountVectorizer from sklearn.model_selection import train_test_split from datetime import datetime -import warnings -warnings.filterwarnings('ignore') - #Functions for splitting into train/test sets. def split_dataset(df, test_size=0.25): movies_train, movies_test = train_test_split(df, test_size=test_size, shuffle=True)