From 7d3951766534e1d1b4ed54265b9dff831765b61b Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Micha=C5=82=20Dudziak?= Date: Wed, 10 May 2023 13:17:38 +0200 Subject: [PATCH] Zaktualizuj 'sacred.py' --- sacred.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/sacred.py b/sacred.py index d9c2c48..42342e9 100644 --- a/sacred.py +++ b/sacred.py @@ -29,7 +29,8 @@ def run_experiment(): cars_train, cars_test = sklearn.model_selection.train_test_split(cars_normalized, test_size=23586, random_state=1) cars_dev, cars_test = sklearn.model_selection.train_test_split(cars_test, test_size=11793, random_state=1) cars_train.rename(columns = {list(cars_train)[0]: 'id'}, inplace = True) - cars_train.to_csv('test.csv') + cars_train.to_csv('train.csv') + cars_test.to_csv('test,csv') feature_cols = ['year', 'mileage', 'vol_engine'] inputs = tf.keras.Input(shape=(len(feature_cols),))