artifacts/scaffold_aware/repeat_2/RUN_METADATA.json

{
  "seed": 123457,
  "device": "cuda",
  "smoke_only": false,
  "smoke_only_untrained_test_lab_indices": [],
  "descriptor_features_predeclared": [
    "LogP",
    "BertzCT",
    "MolMR",
    "NumAromaticRings",
    "HeavyAtomCount"
  ],
  "neural_config": {
    "base_models": [
      "gat",
      "gcn",
      "fpnn"
    ],
    "gat": {
      "model": {
        "graph_model_kwargs": {
          "input_dim": 8,
          "hidden_dim": 320,
          "num_layers": 4,
          "dropout": 0.18,
          "num_labs": 23,
          "lab_embed_dim": 32
        },
        "graph_feature_dim": 320,
        "descriptor_hidden_dims": [
          160,
          80
        ],
        "final_hidden_dims": [
          160,
          80
        ],
        "dropout": 0.2,
        "use_batch_norm": true
      },
      "training": {
        "epochs": 600,
        "lr": 0.0003,
        "weight_decay": 5e-06,
        "patience": 60,
        "plateau_patience": 40,
        "factor": 0.6,
        "min_lr": 1e-06,
        "batch_size": 32,
        "gradient_clip": 1.0
      }
    },
    "gcn": {
      "model": {
        "graph_model_kwargs": {
          "input_dim": 8,
          "hidden_dim": 256,
          "num_layers": 5,
          "dropout": 0.2,
          "num_labs": 23,
          "lab_embed_dim": 32
        },
        "graph_feature_dim": 256,
        "descriptor_hidden_dims": [
          128,
          64
        ],
        "final_hidden_dims": [
          128,
          64
        ],
        "dropout": 0.2,
        "use_batch_norm": true
      },
      "training": {
        "epochs": 600,
        "lr": 0.0004,
        "weight_decay": 1e-05,
        "patience": 60,
        "plateau_patience": 35,
        "factor": 0.6,
        "min_lr": 1e-06,
        "batch_size": 32,
        "gradient_clip": 1.0
      }
    },
    "fpnn": {
      "model": {
        "input_dim": 2048,
        "hidden_dims": [
          768,
          384,
          192,
          96
        ],
        "dropout": 0.15,
        "use_batch_norm": true,
        "num_labs": 23,
        "lab_embed_dim": 64,
        "input_dropout": 0.1
      },
      "training": {
        "epochs": 600,
        "lr": 0.0015,
        "weight_decay": 1e-05,
        "patience": 60,
        "warmup_epochs": 15,
        "batch_size": 128,
        "gradient_clip": 1.0
      }
    },
    "meta_learner": {
      "n_estimators": 800,
      "max_depth": 12,
      "min_samples_split": 5,
      "min_samples_leaf": 3,
      "max_features": 0.8,
      "bootstrap": true,
      "max_samples": 0.85,
      "n_jobs": -1
    },
    "device": "auto"
  },
  "parameter_counts": {
    "gat": 505377,
    "gcn": 391137,
    "fpnn": 2014977
  },
  "training_seconds_by_model": {
    "gat": 2284.689162999988,
    "gcn": 2203.2741107999973,
    "fpnn": 128.70475809999334
  },
  "total_run_seconds": 4665.188108800001,
  "gpu_name": "NVIDIA GeForce RTX 4080",
  "test_set_used_for_selection": false,
  "checkpoint_root": "checkpoints"
}