Skip to content

Repository files navigation

AI Trading Assistant

📈 AI Trading Assistant

An AI-powered stock market analysis application that predicts the probability of a successful trade using Machine Learning, explains predictions with SHAP Explainable AI, and generates market insights using a Large Language Model (LLM).

🌐 Live Demo

Live App: (https://ai-trading-assistant.streamlit.app/)

💻 GitHub Repository

https://github.com/Anam-Daoud/AI-Trading-Assistant


🚀 Features

  • 📊 Live stock market data using Twelve Data API
  • 🤖 Machine Learning trade prediction
  • 📈 Technical indicator generation
  • 🎯 Success probability prediction
  • ⚠️ Risk assessment
  • 🔍 SHAP Explainable AI
  • 🧠 AI-generated market analysis using Groq (Llama 3.3 70B)
  • 📥 Download prediction reports
  • 📥 Download SHAP feature importance reports

🛠 Tech Stack

Programming

  • Python

Machine Learning

  • Scikit-learn
  • Logistic Regression

Data Processing

  • Pandas
  • NumPy

Technical Analysis

  • TA Library

Explainable AI

  • SHAP

Visualization

  • Matplotlib

APIs

  • Twelve Data API
  • Groq API

Deployment

  • Streamlit
  • Streamlit Community Cloud

📂 Project Workflow

Historical Stock Data

Data Cleaning

Exploratory Data Analysis (EDA)

Feature Engineering

Machine Learning Model Training

Model Evaluation

Model Saving

Live Stock Data Collection

Real-Time Prediction

SHAP Explainability

AI Market Analysis

Streamlit Deployment


📊 Prediction Output

The application provides:

  • Trade Prediction
  • Success Probability
  • Risk Level
  • Trading Recommendation
  • SHAP Feature Importance
  • AI-generated Market Analysis

📁 Project Structure

AI-Trading-Assistant/
│
├── app.py
├── trade_model.pkl
├── scaler.pkl
├── requirements.txt
├── README.md
├── historical_data.csv
├── feature_engineered_data.csv
├── model_results.csv
├── shap_feature_importance.csv
├── live_trade_prediction.csv
└── live_shap_feature_importance.csv

⚙️ Installation

pip install -r requirements.txt
streamlit run app.py

📌 Future Improvements

  • Support multiple ML models
  • Interactive stock price charts
  • Portfolio analysis
  • Multi-stock comparison
  • News sentiment analysis
  • Email alerts

👩‍💻 Developed By

Anam Daoud

Software Engineering Student | Machine Learning & AI Enthusiast

About

AI-powered stock trading assistant using Machine Learning, SHAP Explainability, Streamlit, and Groq AI.

Resources

Stars

1 star

Watchers

0 watching

Forks

Releases

Packages

Contributors

Languages