From 376bf22998b7e297e24439e58cb8079b65c9c9d1 Mon Sep 17 00:00:00 2001 From: Serge Retkowsky Date: Fri, 18 Dec 2020 14:41:32 +0100 Subject: [PATCH] Update train.py --- code/train/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/code/train/train.py b/code/train/train.py index 1540be5..c5d44c4 100644 --- a/code/train/train.py +++ b/code/train/train.py @@ -78,7 +78,7 @@ def main(args): X, y = datasets.load_iris(return_X_y=True) #dividing X,y into train and test data - x_train, x_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=223) + x_train, x_test, y_train, y_test = train_test_split(X, y, test_size=0.3, random_state=223) data = {'train': {'X': x_train, 'y': y_train}, 'test': {'X': x_test, 'y': y_test}}