changed the train test split

pgurazada · Jul 22, 2023 · a6f51f2 · a6f51f2
1 parent 59e4051
commit a6f51f2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cli/jobs/pipelines/nyc-taxi/src/train/train.py b/cli/jobs/pipelines/nyc-taxi/src/train/train.py
@@ -73,7 +73,7 @@
 ]
 
 # Split the data into train and test sets
-trainX, testX, trainy, testy = train_test_split(X, y, test_size=0.3, random_state=42)
+trainX, testX, trainy, testy = train_test_split(X, y, test_size=0.2, random_state=42)
 print(trainX.shape)
 print(trainX.columns)