stock-prediction-lstm / config.json
Volko76's picture
Upload config.json with huggingface_hub
701ae5b verified
{
"model_type": "lstm",
"input_size": 8,
"hidden_size": 128,
"num_layers": 2,
"dropout": 0.3,
"sequence_length": 30,
"prediction_days": 7,
"features": [
"Returns",
"High_Low",
"Close_Open",
"MA_5",
"MA_20",
"Volatility",
"Volume_Change",
"RSI"
],
"companies_trained": [
"NVDA",
"AAPL",
"TSLA",
"BYD",
"TTE",
"PLTR",
"AMZN",
"GOOGL",
"MSFT",
"AMD",
"SMCI",
"SNOW"
],
"training_info": {
"epochs": 100,
"batch_size": 32,
"learning_rate": 0.001,
"optimizer": "Adam",
"loss_function": "MSE"
}
}