lstm-stock-prediction-model / stage2_metadata_20250705_170829.json
jengyang's picture
Upload LSTM stock prediction model with sentiment analysis
f4f6611 verified
{
"model_name": "Stage 2 Simple Universal LSTM",
"stocks": [
"AAPL",
"AMZN",
"AVGO",
"BRK.B",
"COST",
"GOOG",
"JNJ",
"JPM",
"LLY",
"MA",
"META",
"MSFT",
"NFLX",
"NVDA",
"ORCL",
"PG",
"TSLA",
"V",
"WMT",
"XOM"
],
"features": [
"Open",
"High",
"Low",
"Close",
"Volume",
"sentiment_10d_avg"
],
"target": "Close",
"time_steps": 60,
"batch_size": 16,
"epochs_trained": 88,
"final_val_loss": 0.0008096768870018423,
"rmse": 26.850322364411017,
"mae": 12.60320354323313,
"r2": 0.9902821175992801,
"mape": 2.679250893651326,
"created_at": "20250705_170829",
"architecture": {
"lstm1_units": 50,
"lstm2_units": 50,
"dropout_rate": 0.2,
"dense_units": 1
}
}