24_May_202202_25PM_1.json

{"model_name": "CustomTransformerDecoder", "n_targets": 2, "model_type": "PyTorch", "model_params": {"n_time_series": 4, "seq_length": 26, "output_seq_length": 1, "output_dim": 2, "n_layers_encoder": 6}, "dataset_params": {"class": "GeneralClassificationLoader", "n_classes": 2, "training_path": "tests/test_data/ff_test.csv", "validation_path": "tests/test_data/ff_test.csv", "test_path": "tests/test_data/ff_test.csv", "sequence_length": 26, "batch_size": 101, "forecast_history": 26, "train_end": 80, "valid_start": 4, "valid_end": 90, "target_col": ["anomalous_rain"], "relevant_cols": ["anomalous_rain", "tmpf", "cfs", "dwpf", "height"], "scaler": "StandardScaler", "interpolate": {"method": "back_forward_generic", "params": {"relevant_columns": ["cfs", "tmpf", "p01m", "dwpf"]}}, "forecast_length": 1}, "training_params": {"criterion": "CrossEntropyLoss", "optimizer": "Adam", "optim_params": {}, "lr": 0.03, "epochs": 4, "batch_size": 100, "shuffle": false}, "GCS": false, "wandb": {"name": "flood_forecast_circleci", "tags": ["dummy_run", "circleci", "multi_head", "classification"], "project": "repo-flood_forecast"}, "forward_params": {}, "metrics": ["CrossEntropyLoss"], "run": [{"epoch": 0, "train_loss": "0.0680028973509454", "validation_loss": "113.27512431330979"}, {"epoch": 1, "train_loss": "0.05458420396058096", "validation_loss": "108.99862844124436"}, {"epoch": 2, "train_loss": "0.054659905693390305", "validation_loss": "106.30307429283857"}, {"epoch": 3, "train_loss": "0.054730736438391936", "validation_loss": "104.98548858333379"}]}