{
    "function": "bin.tabr.main",
    "gpus": [
        "NVIDIA GeForce RTX 5060 Ti"
    ],
    "config": {
        "seed": 8,
        "batch_size": 256,
        "patience": 16,
        "n_epochs": Infinity,
        "context_size": 96,
        "data": {
            "seed": 0,
            "cache": true,
            "path": ":data/california",
            "num_policy": "quantile",
            "cat_policy": null,
            "y_policy": "standard"
        },
        "model": {
            "num_embeddings": null,
            "d_main": 303,
            "context_dropout": 0.23258422826138023,
            "d_multiplier": 2.0,
            "encoder_n_blocks": 0,
            "predictor_n_blocks": 1,
            "mixer_normalization": "auto",
            "dropout0": 0.5508268685197037,
            "dropout1": 0.0,
            "normalization": "LayerNorm",
            "activation": "ReLU"
        },
        "optimizer": {
            "type": "AdamW",
            "lr": 0.000280842117039655,
            "weight_decay": 0.0
        }
    },
    "n_parameters": 832948,
    "prediction_type": null,
    "best_epoch": 35,
    "metrics": {
        "train": {
            "rmse": 0.38320519513733176,
            "mae": 0.24375503108044327,
            "r2": 0.8909803032875061,
            "score": -0.38320519513733176
        },
        "val": {
            "rmse": 0.41819938359784825,
            "mae": 0.2577288023945954,
            "r2": 0.8658679127693176,
            "score": -0.41819938359784825
        },
        "test": {
            "rmse": 0.4075406348524046,
            "mae": 0.25643692991200595,
            "r2": 0.8726264238357544,
            "score": -0.4075406348524046
        }
    },
    "time": "0:00:43",
    "chunk_size": null,
    "eval_batch_size": 32768
}
