Skip to content

Latest commit

 

History

History
19 lines (15 loc) · 1.08 KB

README.md

File metadata and controls

19 lines (15 loc) · 1.08 KB

Trading with Reinforcement Learning and LSTM

Overview

This project leverages Reinforcement Learning and LSTM to develop trading strategy using historical stock data. The environment for trading is built using OpenAI's Gym library, and technical indicators are incorporated using the TA-Lib library. The goal is to train an RL model to optimize trading decisions and maximize portfolio value.

AI Trader Performance PLots

Features

  • Reinforcement Learning: Train a model using Stable-Baselines3's PPO algorithm.
  • Custom Trading Environment: Implemented with OpenAI Gym, featuring custom reward functions and risk management strategies.
  • Technical Indicators: Includes calculations for Moving Averages (MA), MACD, Bollinger Bands, and Momentum, RSI, EMA, ...
  • Risk Management: Dynamic stop-loss and take-profit mechanisms to manage risk.
  • Data Visualization: Plotting of portfolio values, stock holdings, and prices over time

Usage

  1. Clone the repository.
  2. Run pip3 install -r requirements.txt.
  3. Edit config.py.
  4. Run python main.py.