mlarena 0.1.0__tar.gz

mlarena-0.1.0/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2024 Your Name
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE. 

mlarena-0.1.0/PKG-INFO

@@ -0,0 +1,136 @@
+Metadata-Version: 2.3
+Name: mlarena
+Version: 0.1.0
+Summary: An algorithm-agnostic machine learning toolkit for model training, diagnostics and optimization
+License: MIT
+Keywords: machine-learning,data-science,preprocessing,pipeline
+Author: Mena Wang
+Author-email: ningwang25@gmail.com
+Requires-Python: >=3.10
+Classifier: Development Status :: 3 - Alpha
+Classifier: Intended Audience :: Science/Research
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Provides-Extra: demo-dependencies
+Requires-Dist: hyperopt (>=0.2.7)
+Requires-Dist: matplotlib (>=3.4.0)
+Requires-Dist: mlflow (>=2.0.0)
+Requires-Dist: numpy (>=1.21.0)
+Requires-Dist: pandas (>=1.3.0)
+Requires-Dist: scikit-learn (>=0.24.0)
+Requires-Dist: shap (>=0.41.0)
+Project-URL: Homepage, https://github.com/MenaWANG/mlarena
+Project-URL: Repository, https://github.com/MenaWANG/mlarena
+Description-Content-Type: text/markdown
+
+# MLArena
+
+An algorithm-agnostic machine learning toolkit for model training, diagnostics and optimization.
+
+## Features
+
+- **Comprehensive ML Pipeline**:
+  - End-to-end workflow from preprocessing to deployment
+  - Model-agnostic design (works with any scikit-learn compatible model)
+  - Support for both classification and regression tasks
+  - Early stopping and validation set support
+  - MLflow integration for experiment tracking and deployment  
+
+- **Intelligent Preprocessing**:
+  - Automated feature type detection and handling
+  - Smart encoding recommendations based on feature cardinality and rare category
+  - Target encoding with visualization to support smoothing parameter selection
+  - Missing value handling with configurable strategies
+  - Feature selection recommendations with mutual information analysis
+
+- **Advanced Model Evaluation**:
+  - Comprehensive metrics for both classification and regression
+  - Diagnostic visualization of model performance
+  - Threshold analysis for classification tasks
+  - SHAP-based model explanations (global and local)
+  - Cross-validation with variance penalty
+
+- **Hyperparameter Optimization**:
+  - Bayesian optimization with Hyperopt
+  - Cross-validation based tuning
+  - Parallel coordinates visualization for search space analysis
+  - Early stopping to prevent overfitting
+  - Variance penalty to ensure stable solutions
+
+
+## Installation
+
+```bash
+pip install mlarena
+```
+
+## Quick Start
+
+```python
+from mlarena import PreProcessor, ML_PIPELINE
+from sklearn.ensemble import RandomForestClassifier
+
+# Initialize the preprocessor
+preprocessor = PreProcessor(
+    num_impute_strategy='median',
+    cat_impute_strategy='most_frequent'
+)
+
+# Initialize the pipeline
+ml_pipeline = ML_PIPELINE(
+    model = RandomForestClassifier(),
+    preprocessor = preprocessor
+)
+
+# Train the model
+ml_pipeline.fit(X_train, y_train)
+
+# Make predictions
+y_pred = ml_pipeline.predict(X_test)
+
+# Comprehensive Evaluation Report and Visuals
+results = ml_pipeline.evaluate(X_test, y_test)
+
+# Explain the model
+ml_pipeline.explain_model(X_test)
+
+```
+
+## Documentation
+
+### PreProcessor
+
+The `PreProcessor` class handles all data preprocessing tasks:
+
+- Filter Feature Selection
+- Categorical encoding (OneHot, Target)
+- Recommendation of encoding strategy
+- Plot to compare target encoding smoothing parameters
+- Numeric scaling
+- Missing value imputation
+
+### ML_PIPELINE
+
+The `ML_PIPELINE` class provides a complete machine learning workflow:
+
+- Algorithm agnostic model wrapper
+- Support both classification (binary) and regression algorithms
+- Model training and scoring
+- Model global and local explanation
+- Model evaluation with comprehensive reporting and plots
+- Iterative hyperparameter tuning with diagnostic plot
+- Threshold analysis and optimization for classification models
+
+
+## Contributing
+
+Contributions are welcome! Please feel free to submit a Pull Request.
+
+## License
+
+This project is licensed under the MIT License - see the LICENSE file for details. 

mlarena-0.1.0/README.md

@@ -0,0 +1,106 @@
+# MLArena
+
+An algorithm-agnostic machine learning toolkit for model training, diagnostics and optimization.
+
+## Features
+
+- **Comprehensive ML Pipeline**:
+  - End-to-end workflow from preprocessing to deployment
+  - Model-agnostic design (works with any scikit-learn compatible model)
+  - Support for both classification and regression tasks
+  - Early stopping and validation set support
+  - MLflow integration for experiment tracking and deployment  
+
+- **Intelligent Preprocessing**:
+  - Automated feature type detection and handling
+  - Smart encoding recommendations based on feature cardinality and rare category
+  - Target encoding with visualization to support smoothing parameter selection
+  - Missing value handling with configurable strategies
+  - Feature selection recommendations with mutual information analysis
+
+- **Advanced Model Evaluation**:
+  - Comprehensive metrics for both classification and regression
+  - Diagnostic visualization of model performance
+  - Threshold analysis for classification tasks
+  - SHAP-based model explanations (global and local)
+  - Cross-validation with variance penalty
+
+- **Hyperparameter Optimization**:
+  - Bayesian optimization with Hyperopt
+  - Cross-validation based tuning
+  - Parallel coordinates visualization for search space analysis
+  - Early stopping to prevent overfitting
+  - Variance penalty to ensure stable solutions
+
+
+## Installation
+
+```bash
+pip install mlarena
+```
+
+## Quick Start
+
+```python
+from mlarena import PreProcessor, ML_PIPELINE
+from sklearn.ensemble import RandomForestClassifier
+
+# Initialize the preprocessor
+preprocessor = PreProcessor(
+    num_impute_strategy='median',
+    cat_impute_strategy='most_frequent'
+)
+
+# Initialize the pipeline
+ml_pipeline = ML_PIPELINE(
+    model = RandomForestClassifier(),
+    preprocessor = preprocessor
+)
+
+# Train the model
+ml_pipeline.fit(X_train, y_train)
+
+# Make predictions
+y_pred = ml_pipeline.predict(X_test)
+
+# Comprehensive Evaluation Report and Visuals
+results = ml_pipeline.evaluate(X_test, y_test)
+
+# Explain the model
+ml_pipeline.explain_model(X_test)
+
+```
+
+## Documentation
+
+### PreProcessor
+
+The `PreProcessor` class handles all data preprocessing tasks:
+
+- Filter Feature Selection
+- Categorical encoding (OneHot, Target)
+- Recommendation of encoding strategy
+- Plot to compare target encoding smoothing parameters
+- Numeric scaling
+- Missing value imputation
+
+### ML_PIPELINE
+
+The `ML_PIPELINE` class provides a complete machine learning workflow:
+
+- Algorithm agnostic model wrapper
+- Support both classification (binary) and regression algorithms
+- Model training and scoring
+- Model global and local explanation
+- Model evaluation with comprehensive reporting and plots
+- Iterative hyperparameter tuning with diagnostic plot
+- Threshold analysis and optimization for classification models
+
+
+## Contributing
+
+Contributions are welcome! Please feel free to submit a Pull Request.
+
+## License
+
+This project is licensed under the MIT License - see the LICENSE file for details. 

mlarena-0.1.0/mlarena/__init__.py

@@ -0,0 +1,13 @@
+"""
+MLArena - A comprehensive ML pipeline wrapper for scikit-learn compatible models.
+
+This package provides:
+- PreProcessor: Advanced data preprocessing with feature analysis and smart encoding
+- ML_PIPELINE: End-to-end ML pipeline with model training, evaluation, and deployment
+"""
+
+from .preprocessor import PreProcessor
+from .pipeline import ML_PIPELINE
+
+__version__ = "0.1.0"
+__all__ = ["PreProcessor", "ML_PIPELINE"]