From 766034b011f620c07195d9b1ed9819afa46ca347 Mon Sep 17 00:00:00 2001
From: "Mcintosh, Liam S (UG - Computer Science)" <lm00840@surrey.ac.uk>
Date: Mon, 24 May 2021 13:51:58 +0100
Subject: [PATCH] Update build.py
---
build.py | 5 ++---
1 file changed, 2 insertions(+), 3 deletions(-)
diff --git a/build.py b/build.py
index dc83f4b..3e7c174 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)
--
GitLab