natSecLabse / config.json
kristiangnordby's picture
Upload config.json with huggingface_hub
9552f98 verified
{
"model_type": "CyberClassifier",
"input_dim": 768,
"hidden_dims": [
512,
256,
128
],
"output_dim": 1,
"activation": "relu",
"final_activation": "sigmoid",
"metrics": {
"loss": 0.2803977429866791,
"accuracy": 0.883508027066472,
"precision": 0.5713161518788117,
"recall": 0.864543018755582,
"auc": 0.9481742427274251,
"f1": 0.6879886282871357,
"tp": 2904,
"tn": 17073,
"fp": 2179,
"fn": 455,
"specificity": 0.8868169540826927,
"npv": 0.9740415335463258
},
"framework": "pytorch"
}