To receive notifications about scheduled maintenance, please subscribe to the mailing-list gitlab-operations@sympa.ethz.ch. You can subscribe to the mailing-list at https://sympa.ethz.ch

Commit 6f80708a authored by Lukas Wolf's avatar Lukas Wolf
Browse files

move shape logging

parent 51193468
......@@ -56,6 +56,10 @@ class Ensemble_torch:
y_train = y_train[:, 1:]
y_val = y_val[:, 1:]
y_test = y_test[:, 1:]
# Log shapes
logging.info(f"Training data shapes X, y: {X_train.shape, y_train.shape}")
logging.info(f"Test data shapes X, y: {X_test.shape, y_test.shape}")
logging.info(f"Validation data shapes: {X_val.shape, y_val.shape}")
# Create dataloaders
train_dataloader = create_dataloader(X_train, y_train, config['batch_size'], 'train')
validation_dataloader = create_dataloader(X_val, y_val, config['batch_size'], 'val')
......
......@@ -44,8 +44,4 @@ def generate_split(X, y):
X_valid = np.array(X_valid)
y_valid = np.array(y_valid)
logging.info(f"Training data shapes X, y: {X_train.shape, y_train.shape}")
logging.info(f"Test data shapes X, y: {X_test.shape, y_test.shape}")
logging.info(f"Validation data shapes: {X_valid.shape, y_valid.shape}")
return X_train, y_train, X_valid, y_valid, X_test, y_test
\ No newline at end of file
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment