renmove first n rows of traiiner

2021-08-20 11:15:48 +08:00 · 2021-08-20 11:15:48 +08:00 · 4cdef8a535
parent 110e140c67
commit 4cdef8a535
1 changed files with 1 additions and 1 deletions
--- 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)