tra-algorithm 1.0.3__tar.gz → 1.0.4__tar.gz

{tra_algorithm-1.0.3/tra_algorithm.egg-info → tra_algorithm-1.0.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: tra-algorithm
-Version: 1.0.3
+Version: 1.0.4
 Summary: Track/Rail Algorithm (TRA) - A novel machine learning algorithm for dynamic model selection
 Home-page: https://github.com/eswaroy/tra_algorithm
 Author: Dasari Ranga Eswar
@@ -16,22 +16,22 @@ Classifier: Intended Audience :: Science/Research
 Classifier: License :: OSI Approved :: MIT License
 Classifier: Operating System :: OS Independent
 Classifier: Programming Language :: Python :: 3
-Classifier: Programming Language :: Python :: 3.7
 Classifier: Programming Language :: Python :: 3.8
 Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
 Classifier: Topic :: Software Development :: Libraries :: Python Modules
-Requires-Python: >=3.7
+Requires-Python: >=3.8
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: numpy>=1.19.0
-Requires-Dist: pandas>=1.1.0
+Requires-Dist: numpy>=1.21.0
+Requires-Dist: pandas>=1.3.0
 Requires-Dist: scikit-learn>=1.0.0
 Requires-Dist: matplotlib>=3.3.0
 Requires-Dist: joblib>=1.0.0
-Requires-Dist: networkx>=2.5
+Requires-Dist: networkx>=2.6.0
 Provides-Extra: dev
 Requires-Dist: pytest>=6.0; extra == "dev"
 Requires-Dist: pytest-cov>=2.10; extra == "dev"

{tra_algorithm-1.0.3 → tra_algorithm-1.0.4}/docs/CHANGELOG.md

@@ -6,6 +6,26 @@ All notable changes to the TRA Algorithm package will be documented in this file
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
+## [1.0.4] - 2026-03-07
+
+### Added
+- Stronger Router implementations (XGBoost, CatBoost, MLP, LightGBM)
+- Heterogeneous Expert Tracks with specialized models
+- Increased Number of Tracks (5-8+) for better specialization
+- Load Balancing Loss for fair expert utilization
+- Top-K Routing for selective expert usage
+- Expert Capacity Control for improved memory efficiency
+- Router Meta-Features for enhanced routing decisions
+- Temperature-Scaled Soft Routing for better convergence
+- Dynamic Track Creation capabilities
+- Track Specialization via K-Means Clustering
+
+### Improved
+- Algorithm performance significantly enhanced with mixture-of-experts approach
+- Switch Transformer-inspired routing mechanism
+- Signal-Guided Routing with structural signal extraction
+- Memory optimization and computational efficiency
+
 ## [1.0.0] - 2024-12-XX
 
 ### Added

{tra_algorithm-1.0.3 → tra_algorithm-1.0.4}/pyproject.toml

@@ -10,7 +10,7 @@ authors = [
 description = "Track/Rail Algorithm (TRA) - A novel machine learning algorithm for dynamic model selection"
 readme = "README.md"
 license = {text = "MIT"}
-requires-python = ">=3.7"
+requires-python = ">=3.8"
 classifiers = [
     "Development Status :: 4 - Beta",
     "Intended Audience :: Developers",
@@ -18,21 +18,21 @@ classifiers = [
     "License :: OSI Approved :: MIT License",
     "Operating System :: OS Independent",
     "Programming Language :: Python :: 3",
-    "Programming Language :: Python :: 3.7",
     "Programming Language :: Python :: 3.8",
     "Programming Language :: Python :: 3.9",
     "Programming Language :: Python :: 3.10",
     "Programming Language :: Python :: 3.11",
+    "Programming Language :: Python :: 3.12",
     "Topic :: Scientific/Engineering :: Artificial Intelligence",
     "Topic :: Software Development :: Libraries :: Python Modules",
 ]
 dependencies = [
-    "numpy>=1.19.0",
-    "pandas>=1.1.0",
+    "numpy>=1.21.0",
+    "pandas>=1.3.0",
     "scikit-learn>=1.0.0",
     "matplotlib>=3.3.0",
     "joblib>=1.0.0",
-    "networkx>=2.5",
+    "networkx>=2.6.0",
 ]
 dynamic = ["version"]
 

{tra_algorithm-1.0.3 → tra_algorithm-1.0.4}/setup.py

@@ -28,22 +28,22 @@ setup(
         "License :: OSI Approved :: MIT License",
         "Operating System :: OS Independent",
         "Programming Language :: Python :: 3",
-        "Programming Language :: Python :: 3.7",
         "Programming Language :: Python :: 3.8",
         "Programming Language :: Python :: 3.9",
         "Programming Language :: Python :: 3.10",
         "Programming Language :: Python :: 3.11",
+        "Programming Language :: Python :: 3.12",
         "Topic :: Scientific/Engineering :: Artificial Intelligence",
         "Topic :: Software Development :: Libraries :: Python Modules",
     ],
-    python_requires=">=3.7",
+    python_requires=">=3.8",
     install_requires=[
-        "numpy>=1.19.0",
-        "pandas>=1.1.0",
+        "numpy>=1.21.0",
+        "pandas>=1.3.0",
         "scikit-learn>=1.0.0",
         "matplotlib>=3.3.0",
         "joblib>=1.0.0",
-        "networkx>=2.5",
+        "networkx>=2.6.0",
     ],
     extras_require={
         "dev": [

{tra_algorithm-1.0.3 → tra_algorithm-1.0.4}/tests/test_core.py

@@ -5,6 +5,9 @@ import unittest
 import numpy as np
 from sklearn.datasets import make_classification, make_regression
 from sklearn.model_selection import train_test_split
+import sys
+import os
+sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))
 
 from tra_algorithm.core import OptimizedTRA
 
@@ -91,33 +94,27 @@ class TestOptimizedTRA(unittest.TestCase):
         # Test predictions work with feature selection
         y_pred = tra.predict(self.X_test_clf)
         self.assertEqual(len(y_pred), len(self.y_test_clf))
-    
-    def test_statistics_generation(self):
-        """Test statistics generation."""
-        tra = OptimizedTRA(task_type="classification", random_state=42)
-        tra.fit(self.X_train_clf, self.y_train_clf)
-        tra.predict(self.X_test_clf[:10])  # Make some predictions
-        
-        stats = tra.get_track_statistics()
-        
-        # Check required keys
-        required_keys = ['n_tracks', 'n_signals', 'total_predictions', 'track_details']
-        for key in required_keys:
-            self.assertIn(key, stats)
         
-        # Check track details
-        self.assertGreater(stats['n_tracks'], 0)
-        self.assertGreaterEqual(stats['total_predictions'], 0)
-    
-    def test_performance_report(self):
-        """Test performance report generation."""
-        tra = OptimizedTRA(task_type="classification", random_state=42)
-        tra.fit(self.X_train_clf, self.y_train_clf)
-        
-        report = tra.get_performance_report()
-        self.assertIsInstance(report, str)
-        self.assertIn("PERFORMANCE REPORT", report)
-        self.assertIn("TRACK PERFORMANCE DETAILS", report)
+    def test_routing_modes_classification(self):
+        """Test hard and soft routing modes for classification."""
+        for mode in ["hard", "soft"]:
+            tra = OptimizedTRA(task_type="classification", n_tracks=3, routing_mode=mode, random_state=42)
+            tra.fit(self.X_train_clf, self.y_train_clf)
+            
+            y_pred = tra.predict(self.X_test_clf)
+            self.assertEqual(len(y_pred), len(self.y_test_clf))
+            
+            y_proba = tra.predict_proba(self.X_test_clf)
+            self.assertEqual(y_proba.shape[0], len(self.y_test_clf))
+            
+    def test_routing_modes_regression(self):
+        """Test hard and soft routing modes for regression."""
+        for mode in ["hard", "soft"]:
+            tra = OptimizedTRA(task_type="regression", n_tracks=3, routing_mode=mode, random_state=42)
+            tra.fit(self.X_train_reg, self.y_train_reg)
+            
+            y_pred = tra.predict(self.X_test_reg)
+            self.assertEqual(len(y_pred), len(self.y_test_reg))
 
 
 class TestUtils(unittest.TestCase):

{tra_algorithm-1.0.3 → tra_algorithm-1.0.4}/tests/test_utils.py

@@ -5,6 +5,9 @@ import unittest
 import numpy as np
 import tempfile
 import os
+import sys
+
+sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))
 
 from tra_algorithm.core import OptimizedTRA
 from tra_algorithm.utils import validate_input, format_performance_metrics

{tra_algorithm-1.0.3 → tra_algorithm-1.0.4}/tra_algorithm/__init__.py

@@ -52,10 +52,14 @@ from .version import __version__
 from .core import (
     OptimizedTRA,
     Track,
-    Signal,
-    Record,
-    EnhancedSignalCondition
+    EnhancedTRA
 )
+
+try:
+    from .core_enhanced import EnhancedTRA as EnhancedTRANew
+except ImportError:
+    EnhancedTRANew = EnhancedTRA
+
 from .utils import (
     create_example_dataset,
     evaluate_model_performance,
@@ -71,10 +75,8 @@ from .examples import (
 __all__ = [
     # Main classes
     'OptimizedTRA',
+    'EnhancedTRA',  # Enhanced version with MoE improvements
     'Track',
-    'Signal', 
-    'Record',
-    'EnhancedSignalCondition',
     
     # Utilities
     'create_example_dataset',