| Category | Badges |
|---|---|
| Project | |
| CI/CD | |
| Quality | |
| Package | |
| Meta |
AmpyFin is an advanced AI-powered trading system designed to trade within the NASDAQ-100. It leverages machine learning and algorithmic trading strategies to make data-driven investment decisions.
The primary goal of AmpyFin as an open source project is to:
- Democratize Algorithmic Trading: Make proven trading frameworks available for everyone to use freely
- Provide Transparency: Offer insights into how machine learning can be applied to financial markets
- Fill a Gap: Contribute to the open source community where there are few published frameworks for effective trading systems
- Enable Collaboration: Create a platform for traders and developers to build upon and improve together
While there are many theoretical trading models in academic literature, AmpyFin aims to bridge the gap between theory and practice with a working implementation that the community can use, study, and enhance.
For a comprehensive guide on how to use AmpyFin, check out our YouTube tutorial playlist.
📚 Comprehensive documentation is now available! Check out the detailed guides in the docs/ folder:
- Configuration Guide - Complete setup instructions, risk management, and parameter tuning
- Trading Strategies - 100+ TA-Lib indicators, strategy categories, and custom development
- Database Schema - MongoDB collections, SQLite structure, and data management
- API Integration - Broker APIs, data providers, and external services
- Troubleshooting Guide - Common issues, solutions, and debugging techniques
- yfinance: Primary source for historical price data and technical indicators
- MongoDB: Securely stores all data and trading logs for historical analysis
AmpyFin uses a ranked ensemble learning system that dynamically evaluates each strategy's performance and adjusts their influence in the final decision accordingly.
- Mean Reversion: Predicts asset prices will return to their historical average
- Momentum: Capitalizes on prevailing market trends
- Arbitrage: Identifies and exploits price discrepancies between related assets
- AI-Driven Custom Strategies: Continuously refined through machine learning
Each strategy is evaluated based on performance metrics and assigned a weight using a sophisticated ranking algorithm. The system adapts to changing market conditions by prioritizing high-performing strategies.
| Component | Description |
|---|---|
control.py |
Configuration interface for trading parameters |
TradeSim/main.py |
Training and testing environment for strategies |
TradeSim/trading.py |
Executes trades based on algorithmic decisions |
TradeSim/ranking.py |
Evaluates and ranks trading strategies |
strategies/* |
Implementation of various trading algorithms |
utilites/* |
General Utility functions |
dbs/* |
Create SQLite DBs of prices and decisions |
- Python 3.8+
- MongoDB
- TA-Lib
- Clone the repository
git clone https://github.com/AmpyFin/ampyfin.git
cd ampyfin- Install dependencies
pip install -r requirements.txt- Install TA-Lib
TA-Lib is required for technical indicators. Choose one of these installation options:
Option 1: Use pre-built binaries (Recommended for Windows users)
- Download the appropriate wheel file for your Python version from here
- Install it with pip:
pip install <downloaded-wheel-file>.whl
Option 2: Build from source (For Linux/Mac users)
- Follow the instructions on the TA-Lib Python repository
- Set up MongoDB
MongoDB is used for storing trading data and strategy results. Here's how to set it up:
-
Option A: MongoDB Atlas (Cloud-hosted, recommended for beginners)
- Create a free account at MongoDB Atlas
- Create a new cluster (the free tier is sufficient)
- Set up database access (username/password)
- Add your IP address to the network access whitelist
- Get your connection string, which will look like:
mongodb+srv://<username>:<password>@cluster0.mongodb.net/ampyfin
-
Option B: Local MongoDB Installation
- Download and install MongoDB Community Edition
- Start the MongoDB service
- Your connection string will be:
mongodb://localhost:27017/ampyfin
- Configure API keys
You need to sign up for the following services to obtain API keys:
- Alpaca - For trading execution
- Weights & Biases - For experiment tracking
Create a .env file based on the template:
API_KEY = "your_alpaca_api_key"
API_SECRET = "your_alpaca_secret_key"
BASE_URL = "https://paper-api.alpaca.markets" # Paper trading (safe for testing)
# BASE_URL = "https://api.alpaca.markets" # Live trading (uses real money)
WANDB_API_KEY = "your_wandb_api_key"
MONGO_URL = "your_mongo_connection_string"
⚠️ IMPORTANT: The default configuration uses Alpaca's paper trading environment. To switch to live trading (using real money), change the BASE_URL to "https://api.alpaca.markets". Only do this once you've thoroughly tested your strategies and understand the risks involved.
- Run the setup script
python setup.pyThis initializes the database structure required for AmpyFin.
Start the ranking and trading systems in separate terminals:
cd TradeSim
python ranking.py
python trading.pyThese are separate operations that can be performed within the TradeSim module:
- Set the mode in
control.py:
mode = 'train'- Run the training module:
python TradeSim/main.py- Set the mode in
control.py:
mode = 'test'- Run the testing module:
python TradeSim/main.py- Set the mode in
control.py:
mode = 'push'- Push your model to MongoDB:
python TradeSim/main.pyFor live trading, it's recommended to:
- Train the system by running
ranking.pyfor at least two weeks, or - Train using the TradeSim module and push changes to MongoDB before executing trades
This ensures the system has properly ranked strategies before making investment decisions.
Comprehensive documentation is available in the docs/ folder:
- Configuration Guide - Detailed configuration parameters and setup instructions
- Trading Strategies - Complete guide to implemented strategies and TA-Lib indicators
- Database Schema - MongoDB and SQLite database structure documentation
- API Integration - Broker APIs, data providers, and external service integration
- Troubleshooting Guide - Common issues, solutions, and debugging techniques
| Topic | Documentation | Description |
|---|---|---|
| Setup | Configuration Guide | Environment setup, API keys, risk management |
| Strategies | Trading Strategies | 100+ TA-Lib indicators, custom strategy development |
| Data | Database Schema | MongoDB collections, SQLite structure, data flow |
| APIs | API Integration | Alpaca, IBKR, yfinance, Weights & Biases |
| Issues | Troubleshooting Guide | Common problems, solutions, debugging |
Contributions are welcome! Please review the Contributing Guide.
This project is licensed under the MIT License. See the LICENSE file for details.