WinstonAI is an advanced GPU-optimized Reinforcement Learning trading bot for binary options trading. Built with PyTorch and designed to fully utilize modern GPU architecture (RTX 3060 Ti 12GB VRAM), WinstonAI employs sophisticated deep learning techniques for real-time market analysis and trading decisions.
-
๐ง Advanced Deep Learning Architecture
- Deep Reinforcement Learning with Dueling DQN
- Multi-head attention mechanism (16 heads)
- 4-layer LSTM networks for temporal pattern recognition
- Mixed precision training for optimal GPU utilization
-
โก GPU Optimization
- Fully optimized for NVIDIA RTX 3060 Ti (12GB VRAM)
- Mixed precision (FP16/FP32) training
- Tensor core acceleration
- Memory-efficient replay buffer management
-
๐ Advanced Technical Analysis
- 50+ technical indicators (SMA, EMA, MACD, RSI, Bollinger Bands, etc.)
- Real-time market data processing
- Multi-asset support (Forex pairs: EUR/USD, GBP/USD, USD/JPY, etc.)
-
๐ฏ Risk Management
- Configurable stop-loss and take-profit levels
- Maximum daily loss limits
- Position sizing algorithms
- Real-time performance monitoring
-
๐ Live Trading Integration
- Real-time integration with PocketOption API
- Automated trade execution
- Live market data streaming
- Performance logging and analytics
- Python 3.8 or higher
- NVIDIA GPU with CUDA support (recommended: RTX 3060 Ti or better)
- CUDA 11.8 or higher
- 8GB+ VRAM (12GB recommended)
- 16GB+ system RAM
- Clone the repository:
git clone https://github.com/ChipaDevTeam/WinstonAI.git
cd WinstonAI- Create a virtual environment:
python -m venv venv
source venv/bin/activate # On Windows: venv\Scripts\activate- Install dependencies:
pip install -r requirements.txt- Install WinstonAI as a package:
pip install -e .Run the quick start example:
python examples/quickstart.pyOr train with more control:
python examples/train_model.pyAfter installation, you can import WinstonAI in your Python scripts:
from winston_ai import Trainer, Config, LiveTrader
from winston_ai import WinstonAI, AdvancedWinstonAI
from winston_ai.indicators import TechnicalIndicatorsSee examples/ directory for complete usage examples.
WinstonAI/
โโโ winston_ai/ # Main library package
โ โโโ __init__.py # Package initialization
โ โโโ models/ # Neural network models
โ โ โโโ winston_model.py # WinstonAI & AdvancedWinstonAI models
โ โ โโโ attention.py # Multi-head attention mechanism
โ โโโ training/ # Training utilities
โ โ โโโ trainer.py # High-level training orchestration
โ โ โโโ agent.py # DQN agent implementation
โ โ โโโ environment.py # Trading environment simulation
โ โโโ trading/ # Live trading functionality
โ โ โโโ live_trader.py # Live trading interface
โ โโโ indicators/ # Technical analysis
โ โ โโโ technical.py # Technical indicators calculator
โ โโโ utils/ # Utility functions
โ โโโ config.py # Configuration management
โ โโโ device.py # GPU/device management
โ โโโ checkpoints.py # Model checkpoint utilities
โโโ examples/ # Example scripts
โ โโโ quickstart.py # Quick start example
โ โโโ train_model.py # Full training example
โ โโโ use_model.py # Inference example
โ โโโ README.md # Examples documentation
โโโ src/ # Legacy scripts (for reference)
โ โโโ train_gpu_optimized.py # GPU-optimized training script
โ โโโ train_rl_5s.py # RL trainer (5s timeframe)
โ โโโ ultra_live_trading_bot.py # High-performance trading bot
โ โโโ live_trading_bot.py # Standard live trading bot
โ โโโ gpu_monitor.py # GPU monitoring utilities
โโโ data/ # Data directory
โ โโโ configs/ # Configuration files
โ โโโ training_config.json # Training configuration
โ โโโ trading_config.json # Trading configuration
โ โโโ gpu_config.json # GPU settings
โโโ models/ # Saved models (gitignored)
โโโ docs/ # Documentation
โโโ tests/ # Unit tests
โโโ requirements.txt # Python dependencies
โโโ setup.py # Package installation
โโโ README.md # This file
# Install the package
pip install -e .
# Run quick start example
python examples/quickstart.pyfrom winston_ai import Trainer, Config
import pandas as pd
# Load your market data
data = pd.read_csv('your_market_data.csv')
# Ensure data has columns: open, high, low, close, volume
# Configure training
config = Config()
config.update('training',
episodes=1000,
batch_size=512,
learning_rate=0.0001
)
# Create trainer and train
trainer = Trainer(data=data, config=config)
metrics = trainer.train(episodes=1000)
# Plot results
trainer.plot_results('training_results.png')from winston_ai import LiveTrader
import pandas as pd
# Load trained model
trader = LiveTrader(
model_path='models/winston_ai_final.pth',
lookback_window=100
)
# Get recent market data
data = get_recent_market_data() # Your data source
# Make prediction
prediction = trader.predict(data)
print(f"Action: {prediction['action_name']}")
print(f"Confidence: {prediction['confidence']:.2%}")
# Check if should trade
if trader.should_trade(data, min_confidence=0.7):
execute_trade(prediction['action_name'])from winston_ai import WinstonAI, AdvancedWinstonAI
import torch
# Create model
model = AdvancedWinstonAI(
state_size=10000,
action_size=3, # HOLD, CALL, PUT
hidden_size=4096
)
# Use for training or inference
model.eval()
with torch.no_grad():
q_values = model(state_tensor)
action = q_values.argmax().item()For more examples, see the examples/ directory.
WinstonAI uses a sophisticated deep learning architecture:
- Input Layer: 100+ features (technical indicators + market data)
- Feature Extraction: 4-layer LSTM (512 hidden units each)
- Attention Mechanism: Multi-head attention (16 heads, 256 dimensions)
- Value & Advantage Streams: Dueling DQN architecture
- Output Layer: 3 actions (CALL, PUT, HOLD)
- Experience replay buffer (1M transitions)
- Target network with soft updates
- Gradient clipping for stability
- Mixed precision training
- Dynamic learning rate scheduling
{
"device": "cuda",
"mixed_precision": true,
"gradient_checkpointing": true,
"memory_efficient": true
}{
"max_daily_loss": 100,
"stop_loss_percent": 0.02,
"take_profit_percent": 0.04,
"trade_amount": 10
}- Training Speed: ~500-1000 episodes/hour (RTX 3060 Ti)
- Inference Time: <10ms per prediction
- Memory Usage: 8-10GB VRAM during training
- Model Size: ~500MB (full checkpoint)
See README_GPU_OPTIMIZATION.md for detailed information on:
- GPU configuration and setup
- Memory optimization techniques
- Performance tuning
- Troubleshooting common issues
We welcome contributions! Please see CONTRIBUTING.md for details on:
- Code of conduct
- Development setup
- Submitting pull requests
- Coding standards
This project is licensed under the MIT License - see the LICENSE file for details.
IMPORTANT: This software is for educational and research purposes only. Trading binary options and forex carries significant financial risk. The developers are not responsible for any financial losses incurred while using this software. Always:
- Test thoroughly with demo accounts before live trading
- Never trade with money you cannot afford to lose
- Understand the risks of automated trading
- Comply with your local financial regulations
- Use proper risk management strategies
- Built with PyTorch
- Technical indicators powered by ta-lib and ta
- Trading API integration via BinaryOptionsToolsV2
- Historical data from Kaggle
- Issues: GitHub Issues
- Discussions: GitHub Discussions
- Multi-GPU training support
- Enhanced ensemble learning
- Additional trading platforms integration
- Web dashboard for monitoring
- Backtesting framework improvements
- Paper trading mode
- Advanced strategy optimization
Made with โค๏ธ by ChipaDevTeam
Star โญ this repository if you find it useful!