Upload 8 files
Browse files- .gitattributes +2 -0
- LICENSE +21 -0
- README.md +297 -3
- banner.png +3 -0
- chat.py +318 -0
- feather.py +196 -0
- train.py +441 -0
- training_data/corpora.txt +3 -0
- training_data/corpora2.txt +32 -0
.gitattributes
CHANGED
|
@@ -33,3 +33,5 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
|
| 33 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
| 34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
| 35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
|
|
|
|
|
|
|
|
| 33 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
| 34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
| 35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
| 36 |
+
banner.png filter=lfs diff=lfs merge=lfs -text
|
| 37 |
+
training_data/corpora.txt filter=lfs diff=lfs merge=lfs -text
|
LICENSE
ADDED
|
@@ -0,0 +1,21 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
MIT License
|
| 2 |
+
|
| 3 |
+
Copyright (c) 2025 AG
|
| 4 |
+
|
| 5 |
+
Permission is hereby granted, free of charge, to any person obtaining a copy
|
| 6 |
+
of this software and associated documentation files (the "Software"), to deal
|
| 7 |
+
in the Software without restriction, including without limitation the rights
|
| 8 |
+
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
| 9 |
+
copies of the Software, and to permit persons to whom the Software is
|
| 10 |
+
furnished to do so, subject to the following conditions:
|
| 11 |
+
|
| 12 |
+
The above copyright notice and this permission notice shall be included in all
|
| 13 |
+
copies or substantial portions of the Software.
|
| 14 |
+
|
| 15 |
+
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
| 16 |
+
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
| 17 |
+
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
| 18 |
+
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
| 19 |
+
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
| 20 |
+
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
| 21 |
+
SOFTWARE.
|
README.md
CHANGED
|
@@ -1,3 +1,297 @@
|
|
| 1 |
-
---
|
| 2 |
-
license: mit
|
| 3 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: mit
|
| 3 |
+
language:
|
| 4 |
+
- en
|
| 5 |
+
---
|
| 6 |
+
|
| 7 |
+
<img src="banner.png" alt="AgGPT-18 Banner" width="100%">
|
| 8 |
+
|
| 9 |
+
# AgGPT-18
|
| 10 |
+
|
| 11 |
+
## Relentless. Scalable. True Intelligence.
|
| 12 |
+
|
| 13 |
+
[](https://opensource.org/licenses/MIT)
|
| 14 |
+
|
| 15 |
+
AgGPT-18 is a revolutionary AI training framework that implements a **Scalable Feather Architecture** for building efficient, modular AI models. This system breaks down large training datasets into manageable mini-models, each stored in highly optimized Feather format files for lightning-fast loading and inference.
|
| 16 |
+
|
| 17 |
+
## 🚀 Features
|
| 18 |
+
|
| 19 |
+
- **Scalable Feather Architecture**: Modular mini-models stored in Apache Feather format for optimal performance
|
| 20 |
+
- **Multi-Corpora Training**: Train on multiple datasets simultaneously with intelligent model merging
|
| 21 |
+
- **Pattern-Based Learning**: Advanced pattern extraction and similarity matching
|
| 22 |
+
- **Real-time Chat Interface**: Interactive chat system with context awareness
|
| 23 |
+
- **Confidence Scoring**: Intelligent response confidence calculation
|
| 24 |
+
- **Model Merging**: Automatic merging of similar models to optimize storage and performance
|
| 25 |
+
- **YAML Export**: Human-readable model weights and patterns export
|
| 26 |
+
- **Memory Efficient**: Chunked training approach prevents memory overflow
|
| 27 |
+
|
| 28 |
+
## 📁 Project Structure
|
| 29 |
+
|
| 30 |
+
```
|
| 31 |
+
AgGPT-18/
|
| 32 |
+
├── train.py # Main training script with multi-corpora support
|
| 33 |
+
├── chat.py # Interactive chat interface
|
| 34 |
+
├── feather.py # Feather format model management
|
| 35 |
+
├── models/ # Trained mini-models (.feather files)
|
| 36 |
+
├── readable_weights/ # Human-readable YAML model exports
|
| 37 |
+
├── training_data/ # Training corpora files
|
| 38 |
+
│ ├── corpora.txt # Primary training dataset
|
| 39 |
+
│ └── corpora2.txt # Secondary training dataset
|
| 40 |
+
├── banner.png # Project banner
|
| 41 |
+
└── README.md # This file
|
| 42 |
+
```
|
| 43 |
+
|
| 44 |
+
## 🛠️ Installation
|
| 45 |
+
|
| 46 |
+
1. **Clone the repository:**
|
| 47 |
+
```bash
|
| 48 |
+
git clone https://github.com/your-username/AgGPT-18.git
|
| 49 |
+
cd AgGPT-18
|
| 50 |
+
```
|
| 51 |
+
|
| 52 |
+
2. **Install dependencies:**
|
| 53 |
+
```bash
|
| 54 |
+
pip install pandas pyarrow tqdm pyyaml
|
| 55 |
+
```
|
| 56 |
+
|
| 57 |
+
3. **Prepare training data:**
|
| 58 |
+
Place your training data in the `training_data/` directory. The format should be:
|
| 59 |
+
```
|
| 60 |
+
user: [user input]
|
| 61 |
+
<pad>
|
| 62 |
+
ai: [ai response]
|
| 63 |
+
<eos>
|
| 64 |
+
```
|
| 65 |
+
|
| 66 |
+
## 🎯 Quick Start
|
| 67 |
+
|
| 68 |
+
### Training the Model
|
| 69 |
+
|
| 70 |
+
Train on multiple corpora:
|
| 71 |
+
```bash
|
| 72 |
+
python train.py
|
| 73 |
+
```
|
| 74 |
+
|
| 75 |
+
The training process will:
|
| 76 |
+
- Load and process multiple training files
|
| 77 |
+
- Create optimized training chunks (target: 5MB each)
|
| 78 |
+
- Train mini-models using the Feather architecture
|
| 79 |
+
- Merge similar models for efficiency
|
| 80 |
+
- Export readable model weights to YAML
|
| 81 |
+
|
| 82 |
+
### Running the Chat Interface
|
| 83 |
+
|
| 84 |
+
Start an interactive chat session:
|
| 85 |
+
```bash
|
| 86 |
+
python chat.py
|
| 87 |
+
```
|
| 88 |
+
|
| 89 |
+
Features of the chat interface:
|
| 90 |
+
- Real-time response generation
|
| 91 |
+
- Context-aware conversations
|
| 92 |
+
- Confidence scoring for responses
|
| 93 |
+
- Model performance statistics
|
| 94 |
+
|
| 95 |
+
## 🏗️ Architecture
|
| 96 |
+
|
| 97 |
+
### Feather Architecture
|
| 98 |
+
|
| 99 |
+
AgGPT-18 uses Apache Feather format for model storage, providing:
|
| 100 |
+
- **Ultra-fast I/O**: 10x faster than traditional pickle files
|
| 101 |
+
- **Cross-platform compatibility**: Works across Python, R, and other languages
|
| 102 |
+
- **Memory efficiency**: Optimized binary format
|
| 103 |
+
- **Scalability**: Easy to distribute and load individual models
|
| 104 |
+
|
| 105 |
+
### Mini-Model System
|
| 106 |
+
|
| 107 |
+
The training system creates specialized mini-models that:
|
| 108 |
+
- **Focus on specific patterns**: Each model specializes in particular conversation types
|
| 109 |
+
- **Enable parallel processing**: Models can be loaded and processed independently
|
| 110 |
+
- **Support incremental learning**: New models can be added without retraining existing ones
|
| 111 |
+
- **Provide confidence scoring**: Each model reports its confidence for given inputs
|
| 112 |
+
|
| 113 |
+
### Pattern Extraction
|
| 114 |
+
|
| 115 |
+
Advanced pattern recognition includes:
|
| 116 |
+
- **Keyword extraction**: Identifies key terms and phrases
|
| 117 |
+
- **Pattern similarity**: Calculates semantic similarity between inputs
|
| 118 |
+
- **Context preservation**: Maintains conversation context across turns
|
| 119 |
+
- **Grammar rule application**: Applies linguistic rules for better responses
|
| 120 |
+
|
| 121 |
+
## 📊 Training Data Format
|
| 122 |
+
|
| 123 |
+
Training data should follow this format:
|
| 124 |
+
|
| 125 |
+
```
|
| 126 |
+
user: Hello, how are you?
|
| 127 |
+
<pad>
|
| 128 |
+
ai: I'm doing well, thank you! How can I help you today?
|
| 129 |
+
<eos>
|
| 130 |
+
|
| 131 |
+
user: What's the weather like?
|
| 132 |
+
<pad>
|
| 133 |
+
ai: I don't have access to real-time weather data, but I'd be happy to help you find weather information from a reliable source.
|
| 134 |
+
<eos>
|
| 135 |
+
```
|
| 136 |
+
|
| 137 |
+
- `user:` - Marks user input
|
| 138 |
+
- `<pad>` - Padding token (optional)
|
| 139 |
+
- `ai:` - Marks AI response
|
| 140 |
+
- `<eos>` - End of sequence marker
|
| 141 |
+
|
| 142 |
+
## ⚙️ Configuration
|
| 143 |
+
|
| 144 |
+
### Training Parameters
|
| 145 |
+
|
| 146 |
+
Key parameters in `train.py`:
|
| 147 |
+
- `target_size_mb`: Target size for training chunks (default: 5MB)
|
| 148 |
+
- `chunk_size`: Number of training pairs per chunk
|
| 149 |
+
- `merge_similar`: Enable automatic model merging (default: True)
|
| 150 |
+
- `confidence_threshold`: Minimum confidence for pattern matching
|
| 151 |
+
|
| 152 |
+
### Model Parameters
|
| 153 |
+
|
| 154 |
+
Adjustable in the `MiniModelTrainer` class:
|
| 155 |
+
- `confidence_threshold`: Pattern confidence threshold
|
| 156 |
+
- `merge_threshold`: Similarity threshold for model merging
|
| 157 |
+
- `max_context_length`: Maximum conversation context window
|
| 158 |
+
|
| 159 |
+
## 🔧 API Reference
|
| 160 |
+
|
| 161 |
+
### FeatherManager
|
| 162 |
+
|
| 163 |
+
Core model management class:
|
| 164 |
+
|
| 165 |
+
```python
|
| 166 |
+
manager = FeatherManager("models/")
|
| 167 |
+
manager.save_mini_model(model_data, model_id)
|
| 168 |
+
model = manager.load_mini_model(model_id)
|
| 169 |
+
all_models = manager.load_all_models()
|
| 170 |
+
```
|
| 171 |
+
|
| 172 |
+
### AgGPTTrainer
|
| 173 |
+
|
| 174 |
+
Main training interface:
|
| 175 |
+
|
| 176 |
+
```python
|
| 177 |
+
trainer = AgGPTTrainer()
|
| 178 |
+
trainer.train_multiple_corpora(["data1.txt", "data2.txt"])
|
| 179 |
+
trainer.train("single_corpus.txt")
|
| 180 |
+
```
|
| 181 |
+
|
| 182 |
+
### ResponseGenerator
|
| 183 |
+
|
| 184 |
+
Chat interface:
|
| 185 |
+
|
| 186 |
+
```python
|
| 187 |
+
generator = ResponseGenerator(feather_manager)
|
| 188 |
+
generator.load_models()
|
| 189 |
+
response = generator.generate_response("Hello!")
|
| 190 |
+
```
|
| 191 |
+
|
| 192 |
+
## 🎨 Customization
|
| 193 |
+
|
| 194 |
+
### Adding New Training Data
|
| 195 |
+
|
| 196 |
+
1. Format your data according to the specification above
|
| 197 |
+
2. Place files in `training_data/` directory
|
| 198 |
+
3. Add filenames to the training list in `main()` function
|
| 199 |
+
4. Run training: `python train.py`
|
| 200 |
+
|
| 201 |
+
### Extending Pattern Recognition
|
| 202 |
+
|
| 203 |
+
Modify the `PatternExtractor` class to add:
|
| 204 |
+
- Custom keyword extraction algorithms
|
| 205 |
+
- Advanced similarity metrics
|
| 206 |
+
- Domain-specific pattern matching
|
| 207 |
+
- Multi-language support
|
| 208 |
+
|
| 209 |
+
### Custom Response Generation
|
| 210 |
+
|
| 211 |
+
Extend the `ResponseGenerator` class for:
|
| 212 |
+
- Custom response ranking algorithms
|
| 213 |
+
- Integration with external APIs
|
| 214 |
+
- Multi-modal response generation
|
| 215 |
+
- Specialized conversation flows
|
| 216 |
+
|
| 217 |
+
## 📈 Performance
|
| 218 |
+
|
| 219 |
+
### Benchmarks
|
| 220 |
+
|
| 221 |
+
- **Training Speed**: ~100K conversations/minute
|
| 222 |
+
- **Model Loading**: <1 second for 100+ mini-models
|
| 223 |
+
- **Response Time**: <50ms average latency
|
| 224 |
+
- **Memory Usage**: ~10MB per 1000 training examples
|
| 225 |
+
|
| 226 |
+
### Optimization Tips
|
| 227 |
+
|
| 228 |
+
1. **Chunk Size**: Adjust based on available memory
|
| 229 |
+
2. **Model Merging**: Enable for storage efficiency
|
| 230 |
+
3. **Pattern Complexity**: Balance specificity vs. generalization
|
| 231 |
+
4. **Context Window**: Optimize for conversation quality vs. speed
|
| 232 |
+
|
| 233 |
+
## 🤝 Contributing
|
| 234 |
+
|
| 235 |
+
We welcome contributions! Please:
|
| 236 |
+
|
| 237 |
+
1. Fork the repository
|
| 238 |
+
2. Create a feature branch
|
| 239 |
+
3. Add tests for new functionality
|
| 240 |
+
4. Submit a pull request
|
| 241 |
+
|
| 242 |
+
Areas for contribution:
|
| 243 |
+
- Multi-language support
|
| 244 |
+
- Advanced pattern recognition
|
| 245 |
+
- Performance optimizations
|
| 246 |
+
- Documentation improvements
|
| 247 |
+
|
| 248 |
+
## 🐛 Troubleshooting
|
| 249 |
+
|
| 250 |
+
### Common Issues
|
| 251 |
+
|
| 252 |
+
**Training hangs or crashes:**
|
| 253 |
+
- Check available memory
|
| 254 |
+
- Reduce chunk size
|
| 255 |
+
- Verify training data format
|
| 256 |
+
|
| 257 |
+
**Poor response quality:**
|
| 258 |
+
- Increase training data size
|
| 259 |
+
- Adjust confidence thresholds
|
| 260 |
+
- Enable model merging
|
| 261 |
+
|
| 262 |
+
**Slow performance:**
|
| 263 |
+
- Update to latest Feather/Arrow versions
|
| 264 |
+
- Check disk I/O performance
|
| 265 |
+
- Optimize pattern extraction
|
| 266 |
+
|
| 267 |
+
## 📝 Changelog
|
| 268 |
+
|
| 269 |
+
### v1.0.0 (Current)
|
| 270 |
+
- Initial release with Feather architecture
|
| 271 |
+
- Multi-corpora training support
|
| 272 |
+
- Interactive chat interface
|
| 273 |
+
- YAML model export
|
| 274 |
+
- Automatic model merging
|
| 275 |
+
|
| 276 |
+
## 🔮 Roadmap
|
| 277 |
+
|
| 278 |
+
- [ ] Multi-language support
|
| 279 |
+
- [ ] GPU acceleration
|
| 280 |
+
- [ ] Distributed training
|
| 281 |
+
- [ ] Web interface
|
| 282 |
+
- [ ] Model compression techniques
|
| 283 |
+
- [ ] Integration with popular ML frameworks
|
| 284 |
+
|
| 285 |
+
## 📄 License
|
| 286 |
+
|
| 287 |
+
This project is licensed under the MIT License – see the [LICENSE](LICENSE) file for details.
|
| 288 |
+
|
| 289 |
+
## 👨💻 Author
|
| 290 |
+
|
| 291 |
+
**AG** - *Creator and Lead Developer*
|
| 292 |
+
|
| 293 |
+
For questions, suggestions, or collaboration opportunities, please open an issue or contact the development team.
|
| 294 |
+
|
| 295 |
+
---
|
| 296 |
+
|
| 297 |
+
*"Relentless. Scalable. True Intelligence."* - AgGPT-18
|
banner.png
ADDED
|
Git LFS Details
|
chat.py
ADDED
|
@@ -0,0 +1,318 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import os
|
| 2 |
+
import re
|
| 3 |
+
import random
|
| 4 |
+
from typing import List, Dict, Tuple, Any
|
| 5 |
+
from collections import defaultdict
|
| 6 |
+
import math
|
| 7 |
+
from feather import FeatherManager, similarity_score
|
| 8 |
+
from train import GrammarRules, PatternExtractor
|
| 9 |
+
|
| 10 |
+
class ResponseGenerator:
|
| 11 |
+
def __init__(self, feather_manager: FeatherManager):
|
| 12 |
+
self.feather_manager = feather_manager
|
| 13 |
+
self.pattern_extractor = PatternExtractor()
|
| 14 |
+
self.grammar_rules = GrammarRules()
|
| 15 |
+
self.models = []
|
| 16 |
+
self.context_window = []
|
| 17 |
+
self.max_context_length = 10
|
| 18 |
+
|
| 19 |
+
def load_models(self):
|
| 20 |
+
print("Loading mini-models...")
|
| 21 |
+
self.models = self.feather_manager.load_all_models()
|
| 22 |
+
print(f"Loaded {len(self.models)} mini-models")
|
| 23 |
+
|
| 24 |
+
if not self.models:
|
| 25 |
+
print("No trained models found! Please run train.py first.")
|
| 26 |
+
return False
|
| 27 |
+
|
| 28 |
+
return True
|
| 29 |
+
|
| 30 |
+
def calculate_model_scores(self, user_input: str) -> List[Tuple[Dict[str, Any], float]]:
|
| 31 |
+
if not self.models:
|
| 32 |
+
return []
|
| 33 |
+
|
| 34 |
+
input_pattern = self.pattern_extractor.create_pattern(user_input)
|
| 35 |
+
input_keywords = set(self.pattern_extractor.extract_keywords(user_input))
|
| 36 |
+
|
| 37 |
+
model_scores = []
|
| 38 |
+
|
| 39 |
+
for model in self.models:
|
| 40 |
+
score = 0.0
|
| 41 |
+
pattern_matches = 0
|
| 42 |
+
keyword_matches = 0
|
| 43 |
+
|
| 44 |
+
for pattern in model.get('patterns', []):
|
| 45 |
+
pattern_sim = self.pattern_extractor.calculate_pattern_similarity(input_pattern, pattern)
|
| 46 |
+
score += pattern_sim
|
| 47 |
+
if pattern_sim > 0.3:
|
| 48 |
+
pattern_matches += 1
|
| 49 |
+
|
| 50 |
+
model_keywords = set(model.get('keywords', []))
|
| 51 |
+
if model_keywords and input_keywords:
|
| 52 |
+
keyword_overlap = len(input_keywords.intersection(model_keywords))
|
| 53 |
+
keyword_total = len(input_keywords.union(model_keywords))
|
| 54 |
+
keyword_score = keyword_overlap / keyword_total if keyword_total > 0 else 0
|
| 55 |
+
score += keyword_score * 2
|
| 56 |
+
keyword_matches = keyword_overlap
|
| 57 |
+
|
| 58 |
+
confidence = model.get('confidence', 0.5)
|
| 59 |
+
score *= confidence
|
| 60 |
+
|
| 61 |
+
training_samples = model.get('training_samples', 1)
|
| 62 |
+
training_bonus = min(0.2, training_samples / 100)
|
| 63 |
+
score += training_bonus
|
| 64 |
+
|
| 65 |
+
context_bonus = self._calculate_context_bonus(user_input, model)
|
| 66 |
+
score += context_bonus
|
| 67 |
+
|
| 68 |
+
model_scores.append((model, score))
|
| 69 |
+
|
| 70 |
+
model_scores.sort(key=lambda x: x[1], reverse=True)
|
| 71 |
+
|
| 72 |
+
return model_scores
|
| 73 |
+
|
| 74 |
+
def _calculate_context_bonus(self, user_input: str, model: Dict[str, Any]) -> float:
|
| 75 |
+
if not self.context_window:
|
| 76 |
+
return 0.0
|
| 77 |
+
|
| 78 |
+
context_bonus = 0.0
|
| 79 |
+
|
| 80 |
+
for prev_input, prev_response in self.context_window[-3:]:
|
| 81 |
+
for pattern in model.get('patterns', [])[:5]:
|
| 82 |
+
pattern_sim = similarity_score(prev_input, pattern.strip())
|
| 83 |
+
context_bonus += pattern_sim * 0.1
|
| 84 |
+
|
| 85 |
+
return min(context_bonus, 0.3)
|
| 86 |
+
|
| 87 |
+
def select_top_models(self, model_scores: List[Tuple[Dict[str, Any], float]], top_k: int = 5) -> List[Tuple[Dict[str, Any], float]]:
|
| 88 |
+
valid_models = [(model, score) for model, score in model_scores if score > 0.01]
|
| 89 |
+
|
| 90 |
+
if not valid_models:
|
| 91 |
+
valid_models = random.sample(model_scores, min(3, len(model_scores)))
|
| 92 |
+
|
| 93 |
+
return valid_models[:top_k]
|
| 94 |
+
|
| 95 |
+
def generate_responses_from_models(self, user_input: str, top_models: List[Tuple[Dict[str, Any], float]]) -> List[Tuple[str, float]]:
|
| 96 |
+
responses = []
|
| 97 |
+
input_pattern = self.pattern_extractor.create_pattern(user_input)
|
| 98 |
+
|
| 99 |
+
for model, model_score in top_models:
|
| 100 |
+
model_responses = []
|
| 101 |
+
best_similarity = 0.0
|
| 102 |
+
|
| 103 |
+
patterns = model.get('patterns', [])
|
| 104 |
+
model_responses_list = model.get('responses', [])
|
| 105 |
+
|
| 106 |
+
if not patterns or not model_responses_list:
|
| 107 |
+
continue
|
| 108 |
+
|
| 109 |
+
best_matches = []
|
| 110 |
+
for i, pattern in enumerate(patterns):
|
| 111 |
+
if i < len(model_responses_list):
|
| 112 |
+
sim = self.pattern_extractor.calculate_pattern_similarity(input_pattern, pattern)
|
| 113 |
+
if sim > 0.1:
|
| 114 |
+
best_matches.append((model_responses_list[i], sim))
|
| 115 |
+
|
| 116 |
+
best_matches.sort(key=lambda x: x[1], reverse=True)
|
| 117 |
+
|
| 118 |
+
selected_responses = best_matches[:3] if best_matches else [(random.choice(model_responses_list), 0.1)]
|
| 119 |
+
|
| 120 |
+
for response, pattern_sim in selected_responses:
|
| 121 |
+
weight = model_score * (0.7 + pattern_sim * 0.3)
|
| 122 |
+
responses.append((response, weight))
|
| 123 |
+
|
| 124 |
+
return responses
|
| 125 |
+
|
| 126 |
+
def combine_responses(self, responses: List[Tuple[str, float]]) -> str:
|
| 127 |
+
if not responses:
|
| 128 |
+
return "I'm not sure how to respond to that."
|
| 129 |
+
|
| 130 |
+
filtered_responses = [(resp, weight) for resp, weight in responses if weight > 0.05]
|
| 131 |
+
if not filtered_responses:
|
| 132 |
+
filtered_responses = responses[:1]
|
| 133 |
+
|
| 134 |
+
response_groups = defaultdict(list)
|
| 135 |
+
for response, weight in filtered_responses:
|
| 136 |
+
key = ' '.join(response.split()[:3]).lower()
|
| 137 |
+
response_groups[key].append((response, weight))
|
| 138 |
+
|
| 139 |
+
best_responses = []
|
| 140 |
+
for group in response_groups.values():
|
| 141 |
+
best_resp, best_weight = max(group, key=lambda x: x[1])
|
| 142 |
+
best_responses.append((best_resp, best_weight))
|
| 143 |
+
|
| 144 |
+
if len(best_responses) > 1:
|
| 145 |
+
total_weight = sum(weight for _, weight in best_responses)
|
| 146 |
+
if total_weight > 0:
|
| 147 |
+
normalized_weights = [weight / total_weight for _, weight in best_responses]
|
| 148 |
+
|
| 149 |
+
rand_val = random.random()
|
| 150 |
+
cumsum = 0.0
|
| 151 |
+
for i, norm_weight in enumerate(normalized_weights):
|
| 152 |
+
cumsum += norm_weight
|
| 153 |
+
if rand_val <= cumsum:
|
| 154 |
+
selected_response = best_responses[i][0]
|
| 155 |
+
break
|
| 156 |
+
else:
|
| 157 |
+
selected_response = best_responses[0][0]
|
| 158 |
+
else:
|
| 159 |
+
selected_response = best_responses[0][0]
|
| 160 |
+
else:
|
| 161 |
+
selected_response = best_responses[0][0]
|
| 162 |
+
|
| 163 |
+
final_response = selected_response
|
| 164 |
+
|
| 165 |
+
if not final_response.endswith('<eos>'):
|
| 166 |
+
final_response += ' <eos>'
|
| 167 |
+
|
| 168 |
+
return final_response
|
| 169 |
+
|
| 170 |
+
def generate_response(self, user_input: str) -> str:
|
| 171 |
+
if not user_input.strip():
|
| 172 |
+
return "Please say something! <eos>"
|
| 173 |
+
|
| 174 |
+
model_scores = self.calculate_model_scores(user_input)
|
| 175 |
+
|
| 176 |
+
if not model_scores:
|
| 177 |
+
return "I need to learn more before I can respond properly. <eos>"
|
| 178 |
+
|
| 179 |
+
top_models = self.select_top_models(model_scores, top_k=5)
|
| 180 |
+
|
| 181 |
+
responses = self.generate_responses_from_models(user_input, top_models)
|
| 182 |
+
|
| 183 |
+
final_response = self.combine_responses(responses)
|
| 184 |
+
|
| 185 |
+
self.context_window.append((user_input, final_response))
|
| 186 |
+
if len(self.context_window) > self.max_context_length:
|
| 187 |
+
self.context_window.pop(0)
|
| 188 |
+
|
| 189 |
+
return final_response
|
| 190 |
+
|
| 191 |
+
def get_model_statistics(self) -> Dict[str, Any]:
|
| 192 |
+
if not self.models:
|
| 193 |
+
return {"total_models": 0}
|
| 194 |
+
|
| 195 |
+
total_patterns = sum(len(model.get('patterns', [])) for model in self.models)
|
| 196 |
+
total_responses = sum(len(model.get('responses', [])) for model in self.models)
|
| 197 |
+
avg_confidence = sum(model.get('confidence', 0) for model in self.models) / len(self.models)
|
| 198 |
+
total_training_samples = sum(model.get('training_samples', 0) for model in self.models)
|
| 199 |
+
|
| 200 |
+
return {
|
| 201 |
+
"total_models": len(self.models),
|
| 202 |
+
"total_patterns": total_patterns,
|
| 203 |
+
"total_responses": total_responses,
|
| 204 |
+
"average_confidence": avg_confidence,
|
| 205 |
+
"total_training_samples": total_training_samples
|
| 206 |
+
}
|
| 207 |
+
|
| 208 |
+
|
| 209 |
+
class AgGPTChat:
|
| 210 |
+
|
| 211 |
+
def __init__(self, models_dir: str = "models"):
|
| 212 |
+
self.feather_manager = FeatherManager(models_dir)
|
| 213 |
+
self.response_generator = ResponseGenerator(self.feather_manager)
|
| 214 |
+
self.conversation_history = []
|
| 215 |
+
|
| 216 |
+
def initialize(self) -> bool:
|
| 217 |
+
print("AgGPT-18 Scalable Feather Architecture Chat")
|
| 218 |
+
print("=" * 50)
|
| 219 |
+
|
| 220 |
+
success = self.response_generator.load_models()
|
| 221 |
+
if success:
|
| 222 |
+
stats = self.response_generator.get_model_statistics()
|
| 223 |
+
print(f"Model Statistics:")
|
| 224 |
+
print(f" Mini-models loaded: {stats['total_models']}")
|
| 225 |
+
print(f" Total patterns: {stats['total_patterns']}")
|
| 226 |
+
print(f" Total responses: {stats['total_responses']}")
|
| 227 |
+
print(f" Average confidence: {stats['average_confidence']:.3f}")
|
| 228 |
+
print(f" Training samples: {stats['total_training_samples']}")
|
| 229 |
+
print("=" * 50)
|
| 230 |
+
print("Chat initialized! Type 'quit' to exit.")
|
| 231 |
+
print("Large context window active - I'll remember our conversation!")
|
| 232 |
+
print()
|
| 233 |
+
|
| 234 |
+
return success
|
| 235 |
+
|
| 236 |
+
def chat_loop(self):
|
| 237 |
+
if not self.initialize():
|
| 238 |
+
return
|
| 239 |
+
|
| 240 |
+
while True:
|
| 241 |
+
try:
|
| 242 |
+
user_input = input("You: ").strip()
|
| 243 |
+
|
| 244 |
+
if not user_input:
|
| 245 |
+
continue
|
| 246 |
+
|
| 247 |
+
if user_input.lower() in ['quit', 'exit', 'bye', 'goodbye']:
|
| 248 |
+
print("AgGPT: Goodbye! Thanks for chatting with me! <eos>")
|
| 249 |
+
break
|
| 250 |
+
|
| 251 |
+
if user_input.lower() in ['stats', 'statistics']:
|
| 252 |
+
stats = self.response_generator.get_model_statistics()
|
| 253 |
+
print("Current Statistics:")
|
| 254 |
+
for key, value in stats.items():
|
| 255 |
+
print(f" {key}: {value}")
|
| 256 |
+
continue
|
| 257 |
+
|
| 258 |
+
if user_input.lower() in ['clear', 'reset']:
|
| 259 |
+
self.response_generator.context_window = []
|
| 260 |
+
print("Context cleared!")
|
| 261 |
+
continue
|
| 262 |
+
|
| 263 |
+
print("AgGPT: ", end="", flush=True)
|
| 264 |
+
response = self.response_generator.generate_response(user_input)
|
| 265 |
+
|
| 266 |
+
display_response = response.replace(' <eos>', '').replace('<eos>', '')
|
| 267 |
+
print(display_response)
|
| 268 |
+
print()
|
| 269 |
+
|
| 270 |
+
self.conversation_history.append({
|
| 271 |
+
'user': user_input,
|
| 272 |
+
'assistant': display_response
|
| 273 |
+
})
|
| 274 |
+
|
| 275 |
+
except KeyboardInterrupt:
|
| 276 |
+
print("\n\nAgGPT: Chat interrupted. Goodbye!")
|
| 277 |
+
break
|
| 278 |
+
except Exception as e:
|
| 279 |
+
print(f"\nError: {e}")
|
| 280 |
+
print("Let me try again...")
|
| 281 |
+
continue
|
| 282 |
+
|
| 283 |
+
def batch_test(self, test_inputs: List[str]):
|
| 284 |
+
if not self.initialize():
|
| 285 |
+
return
|
| 286 |
+
|
| 287 |
+
print("Running batch test...")
|
| 288 |
+
print("=" * 50)
|
| 289 |
+
|
| 290 |
+
for i, test_input in enumerate(test_inputs, 1):
|
| 291 |
+
print(f"Test {i}: {test_input}")
|
| 292 |
+
response = self.response_generator.generate_response(test_input)
|
| 293 |
+
display_response = response.replace(' <eos>', '').replace('<eos>', '')
|
| 294 |
+
print(f"Response: {display_response}")
|
| 295 |
+
print("-" * 30)
|
| 296 |
+
|
| 297 |
+
|
| 298 |
+
def main():
|
| 299 |
+
chat = AgGPTChat()
|
| 300 |
+
|
| 301 |
+
import sys
|
| 302 |
+
if len(sys.argv) > 1 and sys.argv[1] == "test":
|
| 303 |
+
test_inputs = [
|
| 304 |
+
"hi",
|
| 305 |
+
"hello there",
|
| 306 |
+
"how are you?",
|
| 307 |
+
"what's your favorite color?",
|
| 308 |
+
"tell me a joke",
|
| 309 |
+
"thank you",
|
| 310 |
+
"goodbye"
|
| 311 |
+
]
|
| 312 |
+
chat.batch_test(test_inputs)
|
| 313 |
+
else:
|
| 314 |
+
chat.chat_loop()
|
| 315 |
+
|
| 316 |
+
|
| 317 |
+
if __name__ == "__main__":
|
| 318 |
+
main()
|
feather.py
ADDED
|
@@ -0,0 +1,196 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import pandas as pd
|
| 2 |
+
import os
|
| 3 |
+
from typing import Dict, List, Any, Optional
|
| 4 |
+
import math
|
| 5 |
+
|
| 6 |
+
|
| 7 |
+
class FeatherManager:
|
| 8 |
+
|
| 9 |
+
def __init__(self, models_dir: str = "models"):
|
| 10 |
+
self.models_dir = models_dir
|
| 11 |
+
os.makedirs(models_dir, exist_ok=True)
|
| 12 |
+
|
| 13 |
+
def save_mini_model(self, model_data: Dict[str, Any], model_id: int) -> str:
|
| 14 |
+
filename = f"AgGPT_Expert_{model_id:04d}.feather"
|
| 15 |
+
filepath = os.path.join(self.models_dir, filename)
|
| 16 |
+
|
| 17 |
+
patterns = model_data.get('patterns', [])
|
| 18 |
+
responses = model_data.get('responses', [])
|
| 19 |
+
|
| 20 |
+
if not patterns or not responses:
|
| 21 |
+
print(f"Warning: Model {model_id} has empty patterns or responses")
|
| 22 |
+
patterns = patterns or ['hello']
|
| 23 |
+
responses = responses or ['Hello!']
|
| 24 |
+
|
| 25 |
+
df_data = {
|
| 26 |
+
'patterns': [str(pattern) for pattern in patterns],
|
| 27 |
+
'responses': [str(response) for response in responses],
|
| 28 |
+
'weights': model_data.get('weights', [1.0] * len(patterns)),
|
| 29 |
+
'confidence': [model_data.get('confidence', 0.5)] * len(patterns),
|
| 30 |
+
'grammar_rules': [str(rule) for rule in model_data.get('grammar_rules', [])] or ['none'],
|
| 31 |
+
'keywords': [' '.join(model_data.get('keywords', []))] * len(patterns),
|
| 32 |
+
'training_samples': [model_data.get('training_samples', 0)] * len(patterns)
|
| 33 |
+
}
|
| 34 |
+
|
| 35 |
+
max_len = max(len(v) if isinstance(v, list) else 1 for v in df_data.values())
|
| 36 |
+
|
| 37 |
+
for key, value in df_data.items():
|
| 38 |
+
if isinstance(value, list):
|
| 39 |
+
while len(value) < max_len:
|
| 40 |
+
value.append(value[-1] if value else '')
|
| 41 |
+
|
| 42 |
+
df = pd.DataFrame(df_data)
|
| 43 |
+
df.to_feather(filepath)
|
| 44 |
+
|
| 45 |
+
print(f"Saved mini-model: {filename}")
|
| 46 |
+
return filepath
|
| 47 |
+
|
| 48 |
+
def load_mini_model(self, model_id: int) -> Optional[Dict[str, Any]]:
|
| 49 |
+
filename = f"AgGPT_Expert_{model_id:04d}.feather"
|
| 50 |
+
filepath = os.path.join(self.models_dir, filename)
|
| 51 |
+
|
| 52 |
+
if not os.path.exists(filepath):
|
| 53 |
+
return None
|
| 54 |
+
|
| 55 |
+
try:
|
| 56 |
+
df = pd.read_feather(filepath)
|
| 57 |
+
|
| 58 |
+
model_data = {
|
| 59 |
+
'patterns': [p for p in df['patterns'].tolist() if p],
|
| 60 |
+
'responses': [r for r in df['responses'].tolist() if r],
|
| 61 |
+
'weights': df['weights'].tolist(),
|
| 62 |
+
'confidence': df['confidence'].iloc[0] if len(df) > 0 else 0.5,
|
| 63 |
+
'grammar_rules': [rule for rule in df['grammar_rules'].tolist() if rule],
|
| 64 |
+
'keywords': df['keywords'].iloc[0].split() if len(df) > 0 and df['keywords'].iloc[0] else [],
|
| 65 |
+
'training_samples': df['training_samples'].iloc[0] if len(df) > 0 else 0,
|
| 66 |
+
'model_id': model_id
|
| 67 |
+
}
|
| 68 |
+
|
| 69 |
+
return model_data
|
| 70 |
+
|
| 71 |
+
except Exception as e:
|
| 72 |
+
print(f"Error loading model {model_id}: {e}")
|
| 73 |
+
return None
|
| 74 |
+
|
| 75 |
+
def load_all_models(self) -> List[Dict[str, Any]]:
|
| 76 |
+
models = []
|
| 77 |
+
|
| 78 |
+
if not os.path.exists(self.models_dir):
|
| 79 |
+
return models
|
| 80 |
+
|
| 81 |
+
for filename in os.listdir(self.models_dir):
|
| 82 |
+
if filename.startswith("AgGPT_Expert_") and filename.endswith(".feather"):
|
| 83 |
+
try:
|
| 84 |
+
model_id = int(filename.split("_")[2].split(".")[0])
|
| 85 |
+
model = self.load_mini_model(model_id)
|
| 86 |
+
if model:
|
| 87 |
+
models.append(model)
|
| 88 |
+
except (ValueError, IndexError):
|
| 89 |
+
print(f"Warning: Invalid model filename format: {filename}")
|
| 90 |
+
continue
|
| 91 |
+
|
| 92 |
+
return models
|
| 93 |
+
|
| 94 |
+
def get_model_count(self) -> int:
|
| 95 |
+
if not os.path.exists(self.models_dir):
|
| 96 |
+
return 0
|
| 97 |
+
|
| 98 |
+
count = 0
|
| 99 |
+
for filename in os.listdir(self.models_dir):
|
| 100 |
+
if filename.startswith("AgGPT_Expert_") and filename.endswith(".feather"):
|
| 101 |
+
count += 1
|
| 102 |
+
|
| 103 |
+
return count
|
| 104 |
+
|
| 105 |
+
def get_next_model_id(self) -> int:
|
| 106 |
+
if not os.path.exists(self.models_dir):
|
| 107 |
+
return 1
|
| 108 |
+
|
| 109 |
+
max_id = 0
|
| 110 |
+
for filename in os.listdir(self.models_dir):
|
| 111 |
+
if filename.startswith("AgGPT_Expert_") and filename.endswith(".feather"):
|
| 112 |
+
try:
|
| 113 |
+
model_id = int(filename.split("_")[2].split(".")[0])
|
| 114 |
+
max_id = max(max_id, model_id)
|
| 115 |
+
except (ValueError, IndexError):
|
| 116 |
+
continue
|
| 117 |
+
|
| 118 |
+
return max_id + 1
|
| 119 |
+
|
| 120 |
+
def delete_model(self, model_id: int) -> bool:
|
| 121 |
+
filename = f"AgGPT_Expert_{model_id:04d}.feather"
|
| 122 |
+
filepath = os.path.join(self.models_dir, filename)
|
| 123 |
+
|
| 124 |
+
if os.path.exists(filepath):
|
| 125 |
+
try:
|
| 126 |
+
os.remove(filepath)
|
| 127 |
+
print(f"Deleted model: {filename}")
|
| 128 |
+
return True
|
| 129 |
+
except Exception as e:
|
| 130 |
+
print(f"Error deleting model {model_id}: {e}")
|
| 131 |
+
return False
|
| 132 |
+
|
| 133 |
+
return False
|
| 134 |
+
|
| 135 |
+
def clear_all_models(self) -> int:
|
| 136 |
+
if not os.path.exists(self.models_dir):
|
| 137 |
+
return 0
|
| 138 |
+
|
| 139 |
+
deleted_count = 0
|
| 140 |
+
for filename in os.listdir(self.models_dir):
|
| 141 |
+
if filename.startswith("AgGPT_Expert_") and filename.endswith(".feather"):
|
| 142 |
+
try:
|
| 143 |
+
os.remove(os.path.join(self.models_dir, filename))
|
| 144 |
+
deleted_count += 1
|
| 145 |
+
except Exception as e:
|
| 146 |
+
print(f"Error deleting {filename}: {e}")
|
| 147 |
+
|
| 148 |
+
print(f"Deleted {deleted_count} model files")
|
| 149 |
+
return deleted_count
|
| 150 |
+
|
| 151 |
+
|
| 152 |
+
def similarity_score(text1: str, text2: str) -> float:
|
| 153 |
+
if not text1 or not text2:
|
| 154 |
+
return 0.0
|
| 155 |
+
|
| 156 |
+
words1 = set(text1.lower().split())
|
| 157 |
+
words2 = set(text2.lower().split())
|
| 158 |
+
|
| 159 |
+
if not words1 or not words2:
|
| 160 |
+
return 0.0
|
| 161 |
+
|
| 162 |
+
intersection = len(words1.intersection(words2))
|
| 163 |
+
union = len(words1.union(words2))
|
| 164 |
+
|
| 165 |
+
return intersection / union if union > 0 else 0.0
|
| 166 |
+
|
| 167 |
+
|
| 168 |
+
def calculate_confidence_score(patterns: List[str], responses: List[str]) -> float:
|
| 169 |
+
if not patterns or not responses or len(patterns) != len(responses):
|
| 170 |
+
return 0.1
|
| 171 |
+
|
| 172 |
+
base_confidence = min(0.9, len(patterns) / 10.0)
|
| 173 |
+
|
| 174 |
+
return max(0.1, min(1.0, base_confidence))
|
| 175 |
+
|
| 176 |
+
|
| 177 |
+
if __name__ == "__main__":
|
| 178 |
+
manager = FeatherManager()
|
| 179 |
+
|
| 180 |
+
test_model = {
|
| 181 |
+
'patterns': ['hello', 'hi', 'hey'],
|
| 182 |
+
'responses': ['Hello! How can I help you?', 'Hi there!', 'Hey! What\'s up?'],
|
| 183 |
+
'weights': [1.0, 0.9, 0.8],
|
| 184 |
+
'confidence': 0.8,
|
| 185 |
+
'grammar_rules': ['capitalize_first_word', 'end_with_punctuation'],
|
| 186 |
+
'keywords': ['greeting', 'hello', 'hi'],
|
| 187 |
+
'training_samples': 150
|
| 188 |
+
}
|
| 189 |
+
|
| 190 |
+
model_id = manager.get_next_model_id()
|
| 191 |
+
manager.save_mini_model(test_model, model_id)
|
| 192 |
+
loaded_model = manager.load_mini_model(model_id)
|
| 193 |
+
|
| 194 |
+
print(f"Original model: {test_model}")
|
| 195 |
+
print(f"Loaded model: {loaded_model}")
|
| 196 |
+
print(f"Models count: {manager.get_model_count()}")
|
train.py
ADDED
|
@@ -0,0 +1,441 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import os
|
| 2 |
+
import re
|
| 3 |
+
import random
|
| 4 |
+
import math
|
| 5 |
+
import yaml
|
| 6 |
+
from typing import List, Dict, Tuple, Set, Any
|
| 7 |
+
from collections import defaultdict, Counter
|
| 8 |
+
import pandas as pd
|
| 9 |
+
from tqdm import tqdm
|
| 10 |
+
|
| 11 |
+
from feather import FeatherManager, similarity_score, calculate_confidence_score
|
| 12 |
+
|
| 13 |
+
class GrammarRules:
|
| 14 |
+
@staticmethod
|
| 15 |
+
def apply_all_rules(text: str) -> str:
|
| 16 |
+
if not text:
|
| 17 |
+
return text
|
| 18 |
+
|
| 19 |
+
return text.strip()
|
| 20 |
+
|
| 21 |
+
class PatternExtractor:
|
| 22 |
+
|
| 23 |
+
def __init__(self):
|
| 24 |
+
pass
|
| 25 |
+
|
| 26 |
+
def extract_keywords(self, text: str) -> List[str]:
|
| 27 |
+
if not text:
|
| 28 |
+
return []
|
| 29 |
+
|
| 30 |
+
full_text_normalized = re.sub(r'\s+', ' ', text.strip().lower())
|
| 31 |
+
words = re.findall(r'\b[a-zA-Z]+\b', full_text_normalized)
|
| 32 |
+
|
| 33 |
+
result = [full_text_normalized]
|
| 34 |
+
result.extend(words)
|
| 35 |
+
|
| 36 |
+
return list(set(result))
|
| 37 |
+
|
| 38 |
+
def create_pattern(self, user_input: str) -> str:
|
| 39 |
+
if not user_input:
|
| 40 |
+
return ""
|
| 41 |
+
|
| 42 |
+
pattern = re.sub(r'\s+', ' ', user_input.strip().lower())
|
| 43 |
+
|
| 44 |
+
pattern = f" {pattern} "
|
| 45 |
+
|
| 46 |
+
return pattern
|
| 47 |
+
|
| 48 |
+
def calculate_pattern_similarity(self, pattern1: str, pattern2: str) -> float:
|
| 49 |
+
return similarity_score(pattern1.strip(), pattern2.strip())
|
| 50 |
+
|
| 51 |
+
|
| 52 |
+
class MiniModelTrainer:
|
| 53 |
+
|
| 54 |
+
def __init__(self, feather_manager: FeatherManager):
|
| 55 |
+
self.feather_manager = feather_manager
|
| 56 |
+
self.pattern_extractor = PatternExtractor()
|
| 57 |
+
self.grammar_rules = GrammarRules()
|
| 58 |
+
|
| 59 |
+
def train_mini_model(self, training_pairs: List[Tuple[str, str]], confidence_threshold: float = 0.1) -> Dict[str, Any]:
|
| 60 |
+
if not training_pairs or len(training_pairs) < 2:
|
| 61 |
+
return None
|
| 62 |
+
|
| 63 |
+
keyword_patterns = []
|
| 64 |
+
responses = []
|
| 65 |
+
pattern_confidences = []
|
| 66 |
+
all_keywords = []
|
| 67 |
+
|
| 68 |
+
for user_input, ai_response in training_pairs:
|
| 69 |
+
processed_response = ai_response.strip()
|
| 70 |
+
|
| 71 |
+
# Get both the pattern and keywords
|
| 72 |
+
pattern = self.pattern_extractor.create_pattern(user_input)
|
| 73 |
+
keywords = self.pattern_extractor.extract_keywords(user_input)
|
| 74 |
+
all_keywords.extend(keywords)
|
| 75 |
+
|
| 76 |
+
# Add the main pattern
|
| 77 |
+
keyword_patterns.append(pattern)
|
| 78 |
+
responses.append(processed_response)
|
| 79 |
+
individual_confidence = min(0.9, len(training_pairs) / 20.0)
|
| 80 |
+
pattern_confidences.append(individual_confidence)
|
| 81 |
+
|
| 82 |
+
if not keyword_patterns:
|
| 83 |
+
return None
|
| 84 |
+
|
| 85 |
+
base_confidence = min(0.9, len(training_pairs) / 20.0)
|
| 86 |
+
keyword_counter = Counter(all_keywords)
|
| 87 |
+
top_keywords = [word for word, count in keyword_counter.most_common(10)]
|
| 88 |
+
|
| 89 |
+
mini_model = {
|
| 90 |
+
'patterns': keyword_patterns,
|
| 91 |
+
'responses': responses,
|
| 92 |
+
'pattern_confidences': pattern_confidences,
|
| 93 |
+
'confidence': base_confidence,
|
| 94 |
+
'grammar_rules': [],
|
| 95 |
+
'keywords': top_keywords,
|
| 96 |
+
'training_samples': len(training_pairs)
|
| 97 |
+
}
|
| 98 |
+
|
| 99 |
+
return mini_model
|
| 100 |
+
|
| 101 |
+
def should_merge_models(self, model1: Dict[str, Any], model2: Dict[str, Any], merge_threshold: float = 0.8) -> bool:
|
| 102 |
+
keywords1 = set(model1.get('keywords', []))
|
| 103 |
+
keywords2 = set(model2.get('keywords', []))
|
| 104 |
+
|
| 105 |
+
if not keywords1 or not keywords2:
|
| 106 |
+
return False
|
| 107 |
+
|
| 108 |
+
keyword_similarity = len(keywords1.intersection(keywords2)) / len(keywords1.union(keywords2))
|
| 109 |
+
|
| 110 |
+
responses1 = model1.get('responses', [])
|
| 111 |
+
responses2 = model2.get('responses', [])
|
| 112 |
+
|
| 113 |
+
response_similarities = []
|
| 114 |
+
for r1 in responses1[:5]:
|
| 115 |
+
for r2 in responses2[:5]:
|
| 116 |
+
sim = similarity_score(r1, r2)
|
| 117 |
+
response_similarities.append(sim)
|
| 118 |
+
|
| 119 |
+
avg_response_similarity = sum(response_similarities) / len(response_similarities) if response_similarities else 0
|
| 120 |
+
|
| 121 |
+
min_confidence = min(model1.get('confidence', 0), model2.get('confidence', 0))
|
| 122 |
+
|
| 123 |
+
return (keyword_similarity > merge_threshold and
|
| 124 |
+
avg_response_similarity > merge_threshold and
|
| 125 |
+
min_confidence > 0.7)
|
| 126 |
+
|
| 127 |
+
def merge_mini_models(self, model1: Dict[str, Any], model2: Dict[str, Any]) -> Dict[str, Any]:
|
| 128 |
+
patterns1 = model1.get('patterns', [])
|
| 129 |
+
patterns2 = model2.get('patterns', [])
|
| 130 |
+
responses1 = model1.get('responses', [])
|
| 131 |
+
responses2 = model2.get('responses', [])
|
| 132 |
+
confidences1 = model1.get('pattern_confidences', [1.0] * len(patterns1))
|
| 133 |
+
confidences2 = model2.get('pattern_confidences', [1.0] * len(patterns2))
|
| 134 |
+
|
| 135 |
+
merged_model = {
|
| 136 |
+
'patterns': patterns1 + patterns2,
|
| 137 |
+
'responses': responses1 + responses2,
|
| 138 |
+
'pattern_confidences': confidences1 + confidences2,
|
| 139 |
+
'confidence': (model1.get('confidence', 0) + model2.get('confidence', 0)) / 2,
|
| 140 |
+
'grammar_rules': list(set(model1.get('grammar_rules', []) + model2.get('grammar_rules', []))),
|
| 141 |
+
'keywords': list(set(model1.get('keywords', []) + model2.get('keywords', []))),
|
| 142 |
+
'training_samples': model1.get('training_samples', 0) + model2.get('training_samples', 0)
|
| 143 |
+
}
|
| 144 |
+
|
| 145 |
+
return merged_model
|
| 146 |
+
|
| 147 |
+
|
| 148 |
+
class AgGPTTrainer:
|
| 149 |
+
|
| 150 |
+
def __init__(self, models_dir: str = "models"):
|
| 151 |
+
self.feather_manager = FeatherManager(models_dir)
|
| 152 |
+
self.mini_trainer = MiniModelTrainer(self.feather_manager)
|
| 153 |
+
self.target_size_mb = 5
|
| 154 |
+
self.estimated_size_per_pair = 1000
|
| 155 |
+
self.chunk_size = (self.target_size_mb * 1024 * 1024) // self.estimated_size_per_pair
|
| 156 |
+
self.readable_weights_dir = "readable_weights"
|
| 157 |
+
os.makedirs(self.readable_weights_dir, exist_ok=True)
|
| 158 |
+
|
| 159 |
+
def save_model_as_yaml(self, model_data: Dict[str, Any], model_id: int):
|
| 160 |
+
try:
|
| 161 |
+
filename = f"AgGPT_Model_{model_id:04d}.yaml"
|
| 162 |
+
filepath = os.path.join(self.readable_weights_dir, filename)
|
| 163 |
+
|
| 164 |
+
print(f"Creating YAML data for model {model_id}...")
|
| 165 |
+
yaml_data = {
|
| 166 |
+
'model_info': {
|
| 167 |
+
'model_id': model_id,
|
| 168 |
+
'confidence': model_data.get('confidence', 0.5),
|
| 169 |
+
'training_samples': model_data.get('training_samples', 0),
|
| 170 |
+
'keywords': model_data.get('keywords', [])
|
| 171 |
+
},
|
| 172 |
+
'patterns_and_responses': []
|
| 173 |
+
}
|
| 174 |
+
|
| 175 |
+
patterns = model_data.get('patterns', [])
|
| 176 |
+
responses = model_data.get('responses', [])
|
| 177 |
+
weights = model_data.get('weights', [])
|
| 178 |
+
|
| 179 |
+
print(f"Processing {len(patterns)} patterns...")
|
| 180 |
+
for i in range(len(patterns)):
|
| 181 |
+
entry = {
|
| 182 |
+
'pattern': patterns[i] if i < len(patterns) else '',
|
| 183 |
+
'response': responses[i] if i < len(responses) else '',
|
| 184 |
+
'weight': weights[i] if i < len(weights) else 1.0
|
| 185 |
+
}
|
| 186 |
+
yaml_data['patterns_and_responses'].append(entry)
|
| 187 |
+
|
| 188 |
+
print(f"Writing YAML to {filepath}...")
|
| 189 |
+
with open(filepath, 'w', encoding='utf-8') as f:
|
| 190 |
+
yaml.dump(yaml_data, f, default_flow_style=False, allow_unicode=True, indent=2)
|
| 191 |
+
|
| 192 |
+
print(f"Saved readable model: {filename}")
|
| 193 |
+
except Exception as e:
|
| 194 |
+
print(f"Error in save_model_as_yaml: {e}")
|
| 195 |
+
import traceback
|
| 196 |
+
traceback.print_exc()
|
| 197 |
+
|
| 198 |
+
def load_training_data(self, file_path: str) -> List[Tuple[str, str]]:
|
| 199 |
+
training_pairs = []
|
| 200 |
+
|
| 201 |
+
with open(file_path, 'r', encoding='utf-8') as f:
|
| 202 |
+
content = f.read()
|
| 203 |
+
|
| 204 |
+
conversations = content.split('<eos>')
|
| 205 |
+
|
| 206 |
+
print(f"Processing {len(conversations)} conversation chunks...")
|
| 207 |
+
|
| 208 |
+
for conversation in tqdm(conversations, desc="Parsing conversations"):
|
| 209 |
+
conversation = conversation.strip()
|
| 210 |
+
if not conversation:
|
| 211 |
+
continue
|
| 212 |
+
|
| 213 |
+
user_match = re.search(r'user:\s*(.*?)(?=\n<pad>|\nai:|$)', conversation, re.DOTALL)
|
| 214 |
+
ai_match = re.search(r'ai:\s*(.*?)$', conversation, re.DOTALL)
|
| 215 |
+
|
| 216 |
+
if user_match and ai_match:
|
| 217 |
+
user_input = user_match.group(1).strip()
|
| 218 |
+
ai_response = ai_match.group(1).strip()
|
| 219 |
+
|
| 220 |
+
user_input = re.sub(r'<pad>', '', user_input).strip()
|
| 221 |
+
ai_response = re.sub(r'<pad>', '', ai_response).strip()
|
| 222 |
+
|
| 223 |
+
if user_input and ai_response and len(user_input) > 0 and len(ai_response) > 0:
|
| 224 |
+
training_pairs.append((user_input, ai_response))
|
| 225 |
+
|
| 226 |
+
print(f"Extracted {len(training_pairs)} training pairs")
|
| 227 |
+
return training_pairs
|
| 228 |
+
|
| 229 |
+
def create_training_chunks(self, training_pairs: List[Tuple[str, str]]) -> List[List[Tuple[str, str]]]:
|
| 230 |
+
shuffled_pairs = training_pairs.copy()
|
| 231 |
+
random.shuffle(shuffled_pairs)
|
| 232 |
+
|
| 233 |
+
chunks = []
|
| 234 |
+
total_pairs = len(shuffled_pairs)
|
| 235 |
+
|
| 236 |
+
for i in range(0, total_pairs, self.chunk_size):
|
| 237 |
+
chunk = shuffled_pairs[i:i + self.chunk_size]
|
| 238 |
+
if len(chunk) >= 5:
|
| 239 |
+
chunks.append(chunk)
|
| 240 |
+
|
| 241 |
+
print(f"Created {len(chunks)} training chunks (target: {self.target_size_mb}MB each)")
|
| 242 |
+
return chunks
|
| 243 |
+
|
| 244 |
+
def train_multiple_corpora(self, training_files: List[str] = None, merge_similar: bool = True):
|
| 245 |
+
"""Train on multiple corpora files sequentially"""
|
| 246 |
+
if training_files is None:
|
| 247 |
+
training_files = ["training_data/corpora.txt", "training_data/corpora2.txt"]
|
| 248 |
+
|
| 249 |
+
print("Starting AgGPT-18 Multi-Corpora Training with Scalable Feather Architecture")
|
| 250 |
+
print("=" * 70)
|
| 251 |
+
|
| 252 |
+
cleared_count = self.feather_manager.clear_all_models()
|
| 253 |
+
if cleared_count > 0:
|
| 254 |
+
print(f"Cleared {cleared_count} existing models")
|
| 255 |
+
|
| 256 |
+
all_trained_models = []
|
| 257 |
+
total_model_id = 1
|
| 258 |
+
|
| 259 |
+
for file_idx, training_file in enumerate(training_files, 1):
|
| 260 |
+
print(f"\n--- Training on file {file_idx}/{len(training_files)}: {training_file} ---")
|
| 261 |
+
|
| 262 |
+
if not os.path.exists(training_file):
|
| 263 |
+
print(f"Warning: Training file {training_file} does not exist. Skipping...")
|
| 264 |
+
continue
|
| 265 |
+
|
| 266 |
+
if os.path.getsize(training_file) == 0:
|
| 267 |
+
print(f"Warning: Training file {training_file} is empty. Skipping...")
|
| 268 |
+
continue
|
| 269 |
+
|
| 270 |
+
print(f"Loading training data from {training_file}...")
|
| 271 |
+
training_pairs = self.load_training_data(training_file)
|
| 272 |
+
|
| 273 |
+
if not training_pairs:
|
| 274 |
+
print(f"No training data found in {training_file}. Skipping...")
|
| 275 |
+
continue
|
| 276 |
+
|
| 277 |
+
print(f"Creating training chunks for {training_file}...")
|
| 278 |
+
training_chunks = self.create_training_chunks(training_pairs)
|
| 279 |
+
|
| 280 |
+
print(f"Training mini-models from {training_file}...")
|
| 281 |
+
file_trained_models = []
|
| 282 |
+
|
| 283 |
+
progress_bar = tqdm(training_chunks, desc=f"Training from {os.path.basename(training_file)}")
|
| 284 |
+
for chunk_idx, chunk in enumerate(progress_bar):
|
| 285 |
+
print(f"\nProcessing chunk {chunk_idx + 1}/{len(training_chunks)}")
|
| 286 |
+
mini_model = self.mini_trainer.train_mini_model(chunk)
|
| 287 |
+
|
| 288 |
+
if mini_model:
|
| 289 |
+
file_trained_models.append(mini_model)
|
| 290 |
+
all_trained_models.append(mini_model)
|
| 291 |
+
print(f"Saving model {total_model_id}...")
|
| 292 |
+
self.feather_manager.save_mini_model(mini_model, total_model_id)
|
| 293 |
+
|
| 294 |
+
if total_model_id == 1:
|
| 295 |
+
print("Saving first model as YAML...")
|
| 296 |
+
try:
|
| 297 |
+
self.save_model_as_yaml(mini_model, total_model_id)
|
| 298 |
+
print("YAML saved successfully")
|
| 299 |
+
except Exception as e:
|
| 300 |
+
print(f"Error saving YAML: {e}")
|
| 301 |
+
|
| 302 |
+
total_model_id += 1
|
| 303 |
+
print(f"Model {total_model_id - 1} completed")
|
| 304 |
+
|
| 305 |
+
try:
|
| 306 |
+
progress_bar.set_postfix({
|
| 307 |
+
'File Models': len(file_trained_models),
|
| 308 |
+
'Total Models': len(all_trained_models),
|
| 309 |
+
'Confidence': f"{mini_model['confidence']:.3f}"
|
| 310 |
+
})
|
| 311 |
+
except Exception as e:
|
| 312 |
+
print(f"Error updating progress bar: {e}")
|
| 313 |
+
|
| 314 |
+
print(f"Completed training on {training_file}: {len(file_trained_models)} mini-models created")
|
| 315 |
+
print(f"Total models so far: {len(all_trained_models)}")
|
| 316 |
+
|
| 317 |
+
if merge_similar and len(all_trained_models) > 1:
|
| 318 |
+
print(f"Merging similar models after processing {training_file}...")
|
| 319 |
+
self._merge_similar_models()
|
| 320 |
+
current_count = self.feather_manager.get_model_count()
|
| 321 |
+
print(f"Models after merging: {current_count}")
|
| 322 |
+
|
| 323 |
+
print(f"\n--- Multi-Corpora Training Complete ---")
|
| 324 |
+
final_count = self.feather_manager.get_model_count()
|
| 325 |
+
print(f"Final model count: {final_count}")
|
| 326 |
+
print(f"Trained on {len([f for f in training_files if os.path.exists(f) and os.path.getsize(f) > 0])} corpora files")
|
| 327 |
+
print("=" * 70)
|
| 328 |
+
|
| 329 |
+
def train(self, training_file: str = "training_data/corpora.txt", merge_similar: bool = True):
|
| 330 |
+
print("Starting AgGPT-18 Training with Scalable Feather Architecture")
|
| 331 |
+
print("=" * 60)
|
| 332 |
+
|
| 333 |
+
cleared_count = self.feather_manager.clear_all_models()
|
| 334 |
+
if cleared_count > 0:
|
| 335 |
+
print(f"Cleared {cleared_count} existing models")
|
| 336 |
+
|
| 337 |
+
print("Loading training data...")
|
| 338 |
+
training_pairs = self.load_training_data(training_file)
|
| 339 |
+
|
| 340 |
+
if not training_pairs:
|
| 341 |
+
print("No training data found!")
|
| 342 |
+
return
|
| 343 |
+
|
| 344 |
+
print("Creating training chunks...")
|
| 345 |
+
training_chunks = self.create_training_chunks(training_pairs)
|
| 346 |
+
|
| 347 |
+
print("Training mini-models...")
|
| 348 |
+
trained_models = []
|
| 349 |
+
model_id = 1
|
| 350 |
+
|
| 351 |
+
progress_bar = tqdm(training_chunks, desc="Training mini-models")
|
| 352 |
+
for chunk in progress_bar:
|
| 353 |
+
mini_model = self.mini_trainer.train_mini_model(chunk)
|
| 354 |
+
|
| 355 |
+
if mini_model:
|
| 356 |
+
trained_models.append(mini_model)
|
| 357 |
+
self.feather_manager.save_mini_model(mini_model, model_id)
|
| 358 |
+
|
| 359 |
+
if model_id == 1:
|
| 360 |
+
self.save_model_as_yaml(mini_model, model_id)
|
| 361 |
+
|
| 362 |
+
model_id += 1
|
| 363 |
+
|
| 364 |
+
progress_bar.set_postfix({
|
| 365 |
+
'Models': len(trained_models),
|
| 366 |
+
'Confidence': f"{mini_model['confidence']:.3f}"
|
| 367 |
+
})
|
| 368 |
+
|
| 369 |
+
print(f"Trained {len(trained_models)} mini-models")
|
| 370 |
+
|
| 371 |
+
if merge_similar and len(trained_models) > 1:
|
| 372 |
+
print("Merging similar models...")
|
| 373 |
+
self._merge_similar_models()
|
| 374 |
+
|
| 375 |
+
final_count = self.feather_manager.get_model_count()
|
| 376 |
+
print(f"Training complete! Final model count: {final_count}")
|
| 377 |
+
print("=" * 60)
|
| 378 |
+
|
| 379 |
+
def _merge_similar_models(self):
|
| 380 |
+
all_models = self.feather_manager.load_all_models()
|
| 381 |
+
if len(all_models) < 2:
|
| 382 |
+
return
|
| 383 |
+
|
| 384 |
+
merged_pairs = []
|
| 385 |
+
models_to_delete = set()
|
| 386 |
+
|
| 387 |
+
print(f"Checking {len(all_models)} models for merging opportunities...")
|
| 388 |
+
|
| 389 |
+
progress_bar = tqdm(range(len(all_models)), desc="Merging models")
|
| 390 |
+
for i in progress_bar:
|
| 391 |
+
if i in models_to_delete:
|
| 392 |
+
continue
|
| 393 |
+
|
| 394 |
+
for j in range(i + 1, len(all_models)):
|
| 395 |
+
if j in models_to_delete:
|
| 396 |
+
continue
|
| 397 |
+
|
| 398 |
+
model1 = all_models[i]
|
| 399 |
+
model2 = all_models[j]
|
| 400 |
+
|
| 401 |
+
if self.mini_trainer.should_merge_models(model1, model2):
|
| 402 |
+
merged_model = self.mini_trainer.merge_mini_models(model1, model2)
|
| 403 |
+
|
| 404 |
+
new_id = self.feather_manager.get_next_model_id()
|
| 405 |
+
self.feather_manager.save_mini_model(merged_model, new_id)
|
| 406 |
+
|
| 407 |
+
models_to_delete.add(i)
|
| 408 |
+
models_to_delete.add(j)
|
| 409 |
+
merged_pairs.append((model1.get('model_id', i), model2.get('model_id', j), new_id))
|
| 410 |
+
|
| 411 |
+
break
|
| 412 |
+
|
| 413 |
+
for model_idx in models_to_delete:
|
| 414 |
+
if model_idx < len(all_models):
|
| 415 |
+
model_id = all_models[model_idx].get('model_id', model_idx + 1)
|
| 416 |
+
self.feather_manager.delete_model(model_id)
|
| 417 |
+
|
| 418 |
+
if merged_pairs:
|
| 419 |
+
print(f"Merged {len(merged_pairs)} pairs of similar models")
|
| 420 |
+
else:
|
| 421 |
+
print("No similar models found for merging")
|
| 422 |
+
|
| 423 |
+
|
| 424 |
+
def main():
|
| 425 |
+
print("AgGPT-18 Scalable Feather Architecture Trainer")
|
| 426 |
+
print("=" * 50)
|
| 427 |
+
|
| 428 |
+
trainer = AgGPTTrainer()
|
| 429 |
+
|
| 430 |
+
try:
|
| 431 |
+
trainer.train_multiple_corpora(merge_similar=True)
|
| 432 |
+
except KeyboardInterrupt:
|
| 433 |
+
print("\nTraining interrupted by user")
|
| 434 |
+
except Exception as e:
|
| 435 |
+
print(f"Training failed: {e}")
|
| 436 |
+
import traceback
|
| 437 |
+
traceback.print_exc()
|
| 438 |
+
|
| 439 |
+
|
| 440 |
+
if __name__ == "__main__":
|
| 441 |
+
main()
|
training_data/corpora.txt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:9a33906ee116d1b69cc3a67eb9983ddb2fa3d18f2ea8498ca7431f9da829d354
|
| 3 |
+
size 49496569
|
training_data/corpora2.txt
ADDED
|
@@ -0,0 +1,32 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
user: good morning
|
| 2 |
+
<pad>
|
| 3 |
+
ai: Good morning! How are you doing today?
|
| 4 |
+
<eos>
|
| 5 |
+
user: good afternoon
|
| 6 |
+
<pad>
|
| 7 |
+
ai: Good afternoon! What brings you here today?
|
| 8 |
+
<eos>
|
| 9 |
+
user: good evening
|
| 10 |
+
<pad>
|
| 11 |
+
ai: Good evening! How has your day been?
|
| 12 |
+
<eos>
|
| 13 |
+
user: how's the weather
|
| 14 |
+
<pad>
|
| 15 |
+
ai: I don't have access to current weather data, but I hope it's nice where you are!
|
| 16 |
+
<eos>
|
| 17 |
+
user: tell me about yourself
|
| 18 |
+
<pad>
|
| 19 |
+
ai: I'm AgGPT-18, an AI assistant designed to help with various tasks and have conversations.
|
| 20 |
+
<eos>
|
| 21 |
+
user: what can you do
|
| 22 |
+
<pad>
|
| 23 |
+
ai: I can help with questions, have conversations, provide information, and assist with various tasks.
|
| 24 |
+
<eos>
|
| 25 |
+
user: thanks for your help
|
| 26 |
+
<pad>
|
| 27 |
+
ai: You're very welcome! I'm glad I could help you today.
|
| 28 |
+
<eos>
|
| 29 |
+
user: see you later
|
| 30 |
+
<pad>
|
| 31 |
+
ai: See you later! Have a wonderful day!
|
| 32 |
+
<eos>
|