Upload folder using huggingface_hub
Browse files- config.json +13 -0
- metrics.json +11 -0
- model_info.json +6 -0
- model_params.joblib +3 -0
config.json
ADDED
|
@@ -0,0 +1,13 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"prev_hidden_dims": 16,
|
| 3 |
+
"curr_hidden_dims": 64,
|
| 4 |
+
"time_embedding_dims": 8,
|
| 5 |
+
"lstm_embedding_dims": 128,
|
| 6 |
+
"dropout": 0.2,
|
| 7 |
+
"num_layers": 2,
|
| 8 |
+
"num_heads": 8,
|
| 9 |
+
"batch_size": 8,
|
| 10 |
+
"learning_rate": "1e-4",
|
| 11 |
+
"weight_decay": "1e-4",
|
| 12 |
+
"grad_accum_steps": 16
|
| 13 |
+
}
|
metrics.json
ADDED
|
@@ -0,0 +1,11 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"misc/epoch": 365,
|
| 3 |
+
"misc/patience": 8,
|
| 4 |
+
"train/loss": 0.8899281024932861,
|
| 5 |
+
"val/loss": 0.815583348274231,
|
| 6 |
+
"val/top1_accuracy": 0.7163023352622986,
|
| 7 |
+
"val/top2_accuracy": 0.880464494228363,
|
| 8 |
+
"val/top3_accuracy": 0.9355646371841431,
|
| 9 |
+
"val/mrr": 0.8296687006950378,
|
| 10 |
+
"misc/grad": 66.58882141113281
|
| 11 |
+
}
|
model_info.json
ADDED
|
@@ -0,0 +1,6 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"model_class": "TgnSymDynamic",
|
| 3 |
+
"framework": "jax",
|
| 4 |
+
"timestamp": "20260222_030505",
|
| 5 |
+
"model_name": "tgn-sym-dynamic-rsp"
|
| 6 |
+
}
|
model_params.joblib
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:8272302f931716dbf1286dff20c86607acde355e39efad2d4f50daa0b24eb135
|
| 3 |
+
size 1016760
|