From 4cdef8a535a82da3af1192d319f6561879e06e4a Mon Sep 17 00:00:00 2001 From: Gitea Date: Fri, 20 Aug 2021 11:15:48 +0800 Subject: [PATCH] renmove first n rows of traiiner --- app/prediction_service/trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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)