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}}