Hugging Face's logo

Spaces:
AlgoX
/
mlStocks-pred
Sleeping

App Files Community
AlgoX commited on
Commit
591bc46
·
1 Parent(s): 77a4f6b

init

Browse files
This view is limited to 50 files because it contains too many changes.   See raw diff
Files changed (50) hide show
  1. .gitignore +2 -0
  2. checkpoints/run_20251022_141016/best_model.pt +3 -0
  3. checkpoints/run_20251022_141016/checkpoint_epoch_4.pt +3 -0
  4. checkpoints/run_20251022_141016/checkpoint_epoch_5.pt +3 -0
  5. checkpoints/run_20251022_141016/checkpoint_epoch_6.pt +3 -0
  6. checkpoints/run_20251022_141016/checkpoint_epoch_7.pt +3 -0
  7. checkpoints/run_20251022_141016/checkpoint_epoch_8.pt +3 -0
  8. checkpoints/run_20251022_141016/config.json +16 -0
  9. checkpoints/run_20251022_141016/training_metrics.png +3 -0
  10. checkpoints_classical/run_20251022_160934/ARIMA_model.pkl +3 -0
  11. checkpoints_classical/run_20251022_160934/ElasticNet_model.pkl +3 -0
  12. checkpoints_classical/run_20251022_160934/ExpSmoothing_model.pkl +3 -0
  13. checkpoints_classical/run_20251022_160934/ExtraTrees_model.pkl +3 -0
  14. checkpoints_classical/run_20251022_160934/GradientBoosting_model.pkl +3 -0
  15. checkpoints_classical/run_20251022_160934/Lasso_model.pkl +3 -0
  16. checkpoints_classical/run_20251022_160934/LightGBM_model.pkl +3 -0
  17. checkpoints_classical/run_20251022_160934/RandomForest_model.pkl +3 -0
  18. checkpoints_classical/run_20251022_160934/Ridge_model.pkl +3 -0
  19. checkpoints_classical/run_20251022_160934/SARIMAX_model.pkl +3 -0
  20. checkpoints_classical/run_20251022_160934/SVR_model.pkl +3 -0
  21. checkpoints_classical/run_20251022_160934/XGBoost_model.pkl +3 -0
  22. checkpoints_classical/run_20251022_160934/ablation_feature_importance.csv +6 -0
  23. checkpoints_classical/run_20251022_160934/ablation_feature_importance.png +3 -0
  24. checkpoints_classical/run_20251022_160934/config.json +7 -0
  25. checkpoints_classical/run_20251022_160934/model_comparison.png +3 -0
  26. checkpoints_classical/run_20251022_160934/results_comparison.csv +13 -0
  27. checkpoints_classical/run_20251022_160934/top_model_predictions.png +3 -0
  28. checkpoints_mamba2/run_20251022_142851/best_model.pt +3 -0
  29. checkpoints_mamba2/run_20251022_142851/checkpoint_epoch_40.pt +3 -0
  30. checkpoints_mamba2/run_20251022_142851/checkpoint_epoch_5.pt +3 -0
  31. checkpoints_mamba2/run_20251022_142851/checkpoint_epoch_7.pt +3 -0
  32. checkpoints_mamba2/run_20251022_142851/checkpoint_epoch_8.pt +3 -0
  33. checkpoints_mamba2/run_20251022_142851/checkpoint_epoch_9.pt +3 -0
  34. checkpoints_mamba2/run_20251022_142851/config.json +19 -0
  35. checkpoints_mamba2/run_20251022_142851/training_metrics.png +3 -0
  36. checkpoints_mamba2/run_20251022_155249/best_model.pt +3 -0
  37. checkpoints_mamba2/run_20251022_155249/checkpoint_epoch_5.pt +3 -0
  38. checkpoints_mamba2/run_20251022_155249/checkpoint_epoch_6.pt +3 -0
  39. checkpoints_mamba2/run_20251022_155249/checkpoint_epoch_7.pt +3 -0
  40. checkpoints_mamba2/run_20251022_155249/checkpoint_epoch_8.pt +3 -0
  41. checkpoints_mamba2/run_20251022_155249/checkpoint_epoch_9.pt +3 -0
  42. checkpoints_mamba2/run_20251022_155249/config.json +19 -0
  43. checkpoints_mamba2/run_20251022_155249/training_metrics.png +3 -0
  44. checkpoints_xlstm/run_20251022_145257/best_model.pt +3 -0
  45. checkpoints_xlstm/run_20251022_145257/checkpoint_epoch_30.pt +3 -0
  46. checkpoints_xlstm/run_20251022_145257/checkpoint_epoch_35.pt +3 -0
  47. checkpoints_xlstm/run_20251022_145257/checkpoint_epoch_40.pt +3 -0
  48. checkpoints_xlstm/run_20251022_145257/checkpoint_epoch_45.pt +3 -0
  49. checkpoints_xlstm/run_20251022_145257/checkpoint_epoch_5.pt +3 -0
  50. checkpoints_xlstm/run_20251022_145257/config.json +19 -0
.gitignore CHANGED
@@ -1 +1,3 @@
1
  __pycache__/
 
 
 
1
  __pycache__/
2
+ .pytest_cache
3
+ data/
checkpoints/run_20251022_141016/best_model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:87d82ec32ef4780b74904c35f65222b0e1b84d3f5b40a9bd248d7feb2c9c7e98
3
+ size 3240470
checkpoints/run_20251022_141016/checkpoint_epoch_4.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ea5f24aedf184b6b9a9fcbd36d1bf7df6f67621ad499cccd7a95b0cf00798982
3
+ size 3241782
checkpoints/run_20251022_141016/checkpoint_epoch_5.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:20161c7182b838b0c9acc1d96dda58f7dce68315c80167071a6ae1e37f291069
3
+ size 3241782
checkpoints/run_20251022_141016/checkpoint_epoch_6.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a68513624c4f3821bd6a524534450bef075f261fa130415d7ec4c381dfb33f83
3
+ size 3241782
checkpoints/run_20251022_141016/checkpoint_epoch_7.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:33549c57087c71c879aa3295f16c62e95ab70e87c0dc304666b3726e54328ecc
3
+ size 3241782
checkpoints/run_20251022_141016/checkpoint_epoch_8.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9ef689dd1023f8a5f120f71d7c99461018d05e91ce6692251b6f8c959089b973
3
+ size 3241782
checkpoints/run_20251022_141016/config.json ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_name": "HawkPredictor",
3
+ "seq_length": 20,
4
+ "hidden_size": 128,
5
+ "num_layers": 3,
6
+ "conv_kernel_size": 4,
7
+ "dropout": 0.2,
8
+ "batch_size": 64,
9
+ "num_epochs": 100,
10
+ "learning_rate": 0.001,
11
+ "weight_decay": 1e-05,
12
+ "train_split": 0.8,
13
+ "save_every": 5,
14
+ "save_dir": "./checkpoints",
15
+ "device": "cuda"
16
+ }
checkpoints/run_20251022_141016/training_metrics.png ADDED

Git LFS Details

  • SHA256: 83acdc09bbe4b97c00aaba86fe3db1b0b699959170705b6e75811801a90e1405
  • Pointer size: 131 Bytes
  • Size of remote file: 386 kB
checkpoints_classical/run_20251022_160934/ARIMA_model.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:804ec624945c64aa697fed68000b4599a67455d48b49613e7db91991912f0465
3
+ size 3744453
checkpoints_classical/run_20251022_160934/ElasticNet_model.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:60f17d62b60d56fb678c1837c2096816450fdbcc3304a697ab88d0abfcdd17ae
3
+ size 11248
checkpoints_classical/run_20251022_160934/ExpSmoothing_model.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ee7b5355069df0278892d7a5b133d111eb1e42bcd113ebbcfd7ae8a1d05f8057
3
+ size 68866
checkpoints_classical/run_20251022_160934/ExtraTrees_model.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:08eb818079ae4a993cd5199c718020fcd12400e5cfafda927a68ed006975cab5
3
+ size 263204
checkpoints_classical/run_20251022_160934/GradientBoosting_model.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cd8505b237b09919052907fd8b719f1fb308e241503ac521c165117ad635ffbe
3
+ size 320442
checkpoints_classical/run_20251022_160934/Lasso_model.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:57855f6d509fd048b5b2b10cad5fcd0390da6d2d4d9cf8fcea3f3ff9b0489032
3
+ size 11243
checkpoints_classical/run_20251022_160934/LightGBM_model.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f1b18498c68b9b38d54ab319f28d60d3349e748e59cbcc3b0ca778d8110737be
3
+ size 170744
checkpoints_classical/run_20251022_160934/RandomForest_model.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d8cb18af26e578ad04a7654728391897056d85e2c5d1e5e7ca59c8343ac06297
3
+ size 336917
checkpoints_classical/run_20251022_160934/Ridge_model.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d629b25c5ac3f824389bf504b5958bdc66f7bdfd8d6fccbd96d68dfdfc07ea37
3
+ size 11164
checkpoints_classical/run_20251022_160934/SARIMAX_model.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:14376955aaff520af95f1b5c5df59f300c11aee465083a2045788a63c05667d9
3
+ size 29736961
checkpoints_classical/run_20251022_160934/SVR_model.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:727754f41ae1f2a629a626fe9cf69f31b9888ed7f80f1fc594dcb648b33dba34
3
+ size 22456
checkpoints_classical/run_20251022_160934/XGBoost_model.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6d5efe24aefe8b2c709fee4e202ecb0eb757d210d937d9159ba010b8a9f9ec7c
3
+ size 202709
checkpoints_classical/run_20251022_160934/ablation_feature_importance.csv ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ ,val_rmse,val_mae,val_r2
2
+ Top_10_features,0.0125190210599138,0.009143239116822093,-1.3805699456710783
3
+ Top_20_features,0.012927442885095059,0.009256449304853167,-1.5384315780037014
4
+ Top_50_features,0.011954518379875347,0.008799559869446116,-1.1707228333063915
5
+ Top_100_features,0.012425256925909137,0.008991604707560177,-1.3450438164614926
6
+ Top_1340_features,0.012138133467588944,0.008874818454454578,-1.2379172480586842
checkpoints_classical/run_20251022_160934/ablation_feature_importance.png ADDED

Git LFS Details

  • SHA256: 7801c0b09dffe731acd56c81c9e708c9251497fc5774c2bad84df191bfc2c08d
  • Pointer size: 131 Bytes
  • Size of remote file: 226 kB
checkpoints_classical/run_20251022_160934/config.json ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ {
2
+ "data_path": "/home/aman/code/ml_fr/ml_stocks/data/NIFTY_5_years.csv",
3
+ "seq_length": 20,
4
+ "train_split": 0.8,
5
+ "save_dir": "./checkpoints_classical",
6
+ "target_col": "Daily_Return"
7
+ }
checkpoints_classical/run_20251022_160934/model_comparison.png ADDED

Git LFS Details

  • SHA256: 76a6279be76efe864fb8e4796c0a6b0cfaf47a0d1ee1ebf860a92bd64ef1adfe
  • Pointer size: 131 Bytes
  • Size of remote file: 530 kB
checkpoints_classical/run_20251022_160934/results_comparison.csv ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ,train_mse,train_rmse,train_mae,train_r2,val_mse,val_rmse,val_mae,val_r2
2
+ ARIMA,0.00014452898917474021,0.012022021010410031,0.007792903318647907,-0.00030877428287845454,6.673299598546367e-05,0.008169026624112794,0.006134752582600901,-0.013633206053028868
3
+ Lasso,0.00014448437611511812,0.012020165394665671,0.007789492309237244,0.0,6.682176402980454e-05,0.008174458026670915,0.006141867238772229,-0.01498153810453573
4
+ ElasticNet,0.00014448437611511812,0.012020165394665671,0.007789492309237244,0.0,6.682176402980454e-05,0.008174458026670915,0.006141867238772229,-0.01498153810453573
5
+ SARIMAX,0.00014671008764917046,0.01211239396854191,0.007845520717136657,-0.015404513580617119,6.737789061110245e-05,0.00820840365790465,0.006187347760869252,-0.023428759171833624
6
+ ExpSmoothing,0.00014801766311545162,0.012166250988511276,0.00789618815570349,-0.0244544572592289,6.764238754539729e-05,0.008224499227636737,0.0061892549653926624,-0.02744630508808288
7
+ ExtraTrees,7.977913710047749e-05,0.008931916765200932,0.006712378269745828,0.44783554287618366,6.818143906119145e-05,0.008257205281521796,0.006302270945890863,-0.03563416640186534
8
+ XGBoost,1.1606094359081875e-05,0.003406771838424445,0.002619207816410948,0.9196723225642425,7.601825547259017e-05,0.008718844847374575,0.006713412589920121,-0.15467059249107384
9
+ LightGBM,3.404617369942937e-05,0.005834909913565878,0.004253688118344349,0.7643608629883757,8.579314514533779e-05,0.009262458914636965,0.0070955694459281975,-0.303145160603677
10
+ RandomForest,7.453669818891457e-05,0.008633463858088164,0.006482366232993609,0.4841193200742525,0.00014656353925204926,0.01210634293467888,0.008764490927175841,-1.2262101077388246
11
+ GradientBoosting,1.3621164237089087e-05,0.003690686147193918,0.002976007248120633,0.905725694339183,0.00020557502603557416,0.014337887781523964,0.010249565636651567,-2.1225583333657676
12
+ Ridge,5.35115620156718e-05,0.007315159739586812,0.005624538261631983,0.6296377265522717,0.000219307528586793,0.014809035369894726,0.01216722217263858,-2.3311466094137283
13
+ SVR,0.0006298175389394051,0.02509616582148367,0.022932573178316,-3.3590702045015384,0.0005081148895422926,0.02254140389466221,0.02111886453036676,-6.717952969505901
checkpoints_classical/run_20251022_160934/top_model_predictions.png ADDED

Git LFS Details

  • SHA256: 50de7bccb41b2e4dbedafae0a9b0479b18a6bab0f98d67f12f674542fe12d766
  • Pointer size: 132 Bytes
  • Size of remote file: 1.64 MB
checkpoints_mamba2/run_20251022_142851/best_model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d23af54b956fe9a49f279c5af24235a75e23e61f18ef40c99a07613e7d3cf042
3
+ size 5130742
checkpoints_mamba2/run_20251022_142851/checkpoint_epoch_40.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:81b8771f8e02f1e5101aaca6e112437a584aaaecd460eb24c4771aa141928b43
3
+ size 5133010
checkpoints_mamba2/run_20251022_142851/checkpoint_epoch_5.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2477c3e68d0cc560d848e1bbfd940d197022b965bbba4dcac0e945eaaf164de5
3
+ size 5132758
checkpoints_mamba2/run_20251022_142851/checkpoint_epoch_7.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:49ff347fd8656a22727a95faa4d0f81d5e9494cf8164b22885b4d4b4dd0bb533
3
+ size 5132758
checkpoints_mamba2/run_20251022_142851/checkpoint_epoch_8.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1fc84a90520f7a2b05eaa12a2856a6a7991ef5b4239409785586b94574df7395
3
+ size 5132758
checkpoints_mamba2/run_20251022_142851/checkpoint_epoch_9.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4e7d49eb89b92f98ace4e65793af701f6e6e0a39e0ddbf857cf26cded1c8e7b2
3
+ size 5132758
checkpoints_mamba2/run_20251022_142851/config.json ADDED
@@ -0,0 +1,19 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_name": "Mamba2Predictor",
3
+ "seq_length": 20,
4
+ "hidden_size": 128,
5
+ "num_layers": 3,
6
+ "inner_size": null,
7
+ "head_size": 64,
8
+ "bc_head_size": 128,
9
+ "conv_kernel_size": 4,
10
+ "dropout": 0.2,
11
+ "batch_size": 64,
12
+ "num_epochs": 100,
13
+ "learning_rate": 0.001,
14
+ "weight_decay": 1e-05,
15
+ "train_split": 0.8,
16
+ "save_every": 5,
17
+ "save_dir": "./checkpoints_mamba2",
18
+ "device": "cuda"
19
+ }
checkpoints_mamba2/run_20251022_142851/training_metrics.png ADDED

Git LFS Details

  • SHA256: 9254802303629db276aac80994f1d5fdcee05ade2b833d699d89812532c6bdfe
  • Pointer size: 131 Bytes
  • Size of remote file: 357 kB
checkpoints_mamba2/run_20251022_155249/best_model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3d140cb1b09459c48dcc008bec21f9a8ed25ed4e4a24354540480b897f74a501
3
+ size 5130742
checkpoints_mamba2/run_20251022_155249/checkpoint_epoch_5.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:22e97b6dbe694125969295aebb5697942e00f9e8dbcd4026976513fc6f42bb55
3
+ size 5132758
checkpoints_mamba2/run_20251022_155249/checkpoint_epoch_6.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:745adf6c98ef9f44dca7c7d99e8aad6331243fae31d7c7723bda71435d9516f0
3
+ size 5132758
checkpoints_mamba2/run_20251022_155249/checkpoint_epoch_7.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8b2fbb34d5833d5ae1e04e1864eae41942c5d4f716b7037abac2ff2886ab3a6b
3
+ size 5132758
checkpoints_mamba2/run_20251022_155249/checkpoint_epoch_8.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:48add1dbe80ccc68c5feeee66a13417e423d05897956a8a6e36798b42fb9816e
3
+ size 5132758
checkpoints_mamba2/run_20251022_155249/checkpoint_epoch_9.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bc21e3afe820f23dc935b1eb9c0ac4fc4b242f05ab49c33b35c1febc83756caa
3
+ size 5132758
checkpoints_mamba2/run_20251022_155249/config.json ADDED
@@ -0,0 +1,19 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_name": "Mamba2Predictor",
3
+ "seq_length": 20,
4
+ "hidden_size": 128,
5
+ "num_layers": 3,
6
+ "inner_size": null,
7
+ "head_size": 64,
8
+ "bc_head_size": 128,
9
+ "conv_kernel_size": 4,
10
+ "dropout": 0.2,
11
+ "batch_size": 64,
12
+ "num_epochs": 100,
13
+ "learning_rate": 0.001,
14
+ "weight_decay": 1e-05,
15
+ "train_split": 0.8,
16
+ "save_every": 5,
17
+ "save_dir": "./checkpoints_mamba2",
18
+ "device": "cuda"
19
+ }
checkpoints_mamba2/run_20251022_155249/training_metrics.png ADDED

Git LFS Details

  • SHA256: 3cf42194448dcd848ffadfc8eab7a1d55e71e3249301d1d4dc5c14ac71531101
  • Pointer size: 131 Bytes
  • Size of remote file: 280 kB
checkpoints_xlstm/run_20251022_145257/best_model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b49c3a54563aea13d7c93c1e3e104f4a53882cd823b9273d0de98d05e89bbd08
3
+ size 4359262
checkpoints_xlstm/run_20251022_145257/checkpoint_epoch_30.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d622fa1aaa0636af835e0c922fd23712e3cdcd477b3b176b24cd287e3f4a9610
3
+ size 4361314
checkpoints_xlstm/run_20251022_145257/checkpoint_epoch_35.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5b850db5fa8363f30b0e1ce4ebab7809ecd5174476a0410b88d71e9d237a1f58
3
+ size 4361314
checkpoints_xlstm/run_20251022_145257/checkpoint_epoch_40.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:689506e66dfec12fe32d2c025ab266a2cb70230cbfd39d184ba986d321f0ebd7
3
+ size 4361314
checkpoints_xlstm/run_20251022_145257/checkpoint_epoch_45.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8a9e2255dbc6a632961c158ad3a535d174ea04961cea4eed43d698908f222873
3
+ size 4361314
checkpoints_xlstm/run_20251022_145257/checkpoint_epoch_5.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7b48fb497c297267c112242faa298d9c7c2e567203e5d06b0c12406bc7597550
3
+ size 4361086
checkpoints_xlstm/run_20251022_145257/config.json ADDED
@@ -0,0 +1,19 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_name": "xLSTM_mLSTM",
3
+ "block_type": "mlstm",
4
+ "seq_length": 20,
5
+ "hidden_size": 128,
6
+ "num_layers": 3,
7
+ "num_heads": 8,
8
+ "conv_kernel_size": 4,
9
+ "expand_factor": 2,
10
+ "dropout": 0.2,
11
+ "batch_size": 64,
12
+ "num_epochs": 100,
13
+ "learning_rate": 0.001,
14
+ "weight_decay": 1e-05,
15
+ "train_split": 0.8,
16
+ "save_every": 5,
17
+ "save_dir": "./checkpoints_xlstm",
18
+ "device": "cuda"
19
+ }