mlStocks-pred / deployment /models /hawk_config.json
AlgoX's picture
fix
40d1b97
raw
history blame contribute delete
373 Bytes
{
"input_size": 67,
"model_name": "HawkPredictor",
"seq_length": 20,
"hidden_size": 128,
"num_layers": 3,
"conv_kernel_size": 4,
"dropout": 0.2,
"batch_size": 64,
"num_epochs": 100,
"learning_rate": 0.001,
"weight_decay": 1e-05,
"train_split": 0.8,
"save_every": 5,
"save_dir": "./checkpoints",
"device": "cuda"
}