Remove extra validation set (#80)

soda-inria · Nov 27, 2024 · ddc2f35 · ddc2f35
1 parent 7af0407
commit ddc2f35
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/examples/plot_01_survival_analysis.py b/examples/plot_01_survival_analysis.py
@@ -48,8 +48,7 @@
 # %%
 from sklearn.model_selection import train_test_split
 
-X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2)
-X_train, X_val, y_train, y_val = train_test_split(X_train, y_train, test_size=0.2)
+X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.3)
 
 # %%
 #