Update README for #52

EpistasisLab · Dec 8, 2015 · fdd8b37 · fdd8b37
1 parent 5620a70
commit fdd8b37
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -173,7 +173,7 @@ from sklearn.cross_validation import train_test_split
 
 digits = load_digits()
 X_train, X_test, y_train, y_test = train_test_split(digits.data, digits.target,
-                                                    train_size=0.75)
+                                                    train_size=0.75, test_size=0.25)
 
 tpot = TPOT(generations=5)
 tpot.fit(X_train, y_train)
@@ -192,7 +192,7 @@ from sklearn.linear_model import LogisticRegression
 
 # NOTE: Make sure that the class is labeled 'class' in the data file
 tpot_data = pd.read_csv('PATH/TO/DATA/FILE', sep='COLUMN_SEPARATOR')
-training_indeces, testing_indeces = next(iter(StratifiedShuffleSplit(tpot_data['class'].values, n_iter=1, train_size=0.75)))
+training_indeces, testing_indeces = next(iter(StratifiedShuffleSplit(tpot_data['class'].values, n_iter=1, train_size=0.75, test_size=0.25)))
 
 
 result1 = tpot_data.copy()