diff --git a/app/prediction_service/trainer.py b/app/prediction_service/trainer.py index 130925b..000d18d 100644 --- a/app/prediction_service/trainer.py +++ b/app/prediction_service/trainer.py @@ -83,7 +83,7 @@ class Trainer(): factors: the list of factors for training the model ''' try: - df = pd.read_csv(dataPath, nrows=100000) + df = pd.read_csv(dataPath) Y = df['merged_or_not'] X = df[factors] X_train, X_test, Y_train, Y_test = model_selection.train_test_split(X, Y, test_size=0.1, random_state=10)