{
  "model_repo_id": "kyLELEng/market-gpt-return-token-v2",
  "dataset_id": "siddharthmb/stocks-ohlcv",
  "dataset_file": "ohlcv.csv",
  "use_synthetic": false,
  "synthetic_steps": 900,
  "tickers": [
    "AAPL",
    "MSFT",
    "AMZN",
    "GOOGL",
    "NVDA",
    "TSLA",
    "AMD",
    "INTC",
    "JPM",
    "BAC",
    "V",
    "MA"
  ],
  "num_assets": 12,
  "num_buckets": 101,
  "bucket_method": "quantile",
  "clip_value": 5.0,
  "vol_span": 60,
  "context_steps": 128,
  "prediction_steps": 32,
  "window_stride": 1,
  "train_fraction": 0.8,
  "validation_fraction": 0.1,
  "n_embd": 384,
  "n_layer": 6,
  "n_head": 6,
  "dropout": 0.1,
  "forecast_only_loss": true,
  "per_device_train_batch_size": 4,
  "per_device_eval_batch_size": 4,
  "gradient_accumulation_steps": 8,
  "learning_rate": 0.0003,
  "weight_decay": 0.1,
  "warmup_ratio": 0.05,
  "num_train_epochs": 5.0,
  "max_steps": -1,
  "logging_steps": 20,
  "eval_strategy": "epoch",
  "save_strategy": "no",
  "bf16": true,
  "fp16": false,
  "random_seed": 7,
  "num_eval_contexts": 12,
  "num_sample_paths": 24,
  "generation_temperature": 1.0,
  "generation_top_p": 0.95,
  "max_csv_chunks": null,
  "csv_chunksize": 500000,
  "min_rows_per_ticker": 1000,
  "output_dir": "market-gpt-return-token"
}