diff --git a/run.py b/run.py index f6e67d8..487778a 100644 --- a/run.py +++ b/run.py @@ -11,7 +11,7 @@ train = pd.read_csv( on_bad_lines='skip', header=None, quoting=csv.QUOTE_NONE, - nrows=50000) + nrows=40000) labels = pd.read_csv( @@ -20,7 +20,7 @@ labels = pd.read_csv( on_bad_lines='skip', header=None, quoting=csv.QUOTE_NONE, - nrows=50000) + nrows=40000) def data_preprocessing(text):