Metadata-Version: 2.1
Name: aion_optimizer
Version: 0.1.0
Summary: An automated machine learning optimization package
Home-page: https://github.com/
Author: Lalit Lohani
Author-email: lalitlohani01@gmail.com
Classifier: Programming Language :: Python :: 3
Classifier: License :: OSI Approved :: MIT License
Classifier: Operating System :: OS Independent
Requires-Python: >=3.8
Description-Content-Type: text/markdown
License-File: LICENCE
Requires-Dist: numpy>=1.19.0
Requires-Dist: pandas>=1.2.0
Requires-Dist: scikit-learn>=0.24.0
Requires-Dist: scipy
Requires-Dist: xgboost
Requires-Dist: catboost

# Aion Optimizer

A Python package for automated machine learning optimization.

## Installation

```bash
pip install aion_optimizer
```

## Usage

```python
from aion_ml_optimizer import EnhancedSequentialOptimizer
import pandas as pd
from sklearn.model_selection import train_test_split

dataset_path = "..."

df = pd.read_csv(dataset_path)

optimizer = EnhancedSequentialOptimizer(
        task_type='classification',
        target_column='class', 
        generations=5,
        cv_folds=5,
        random_state=42,
        n_jobs=-1,
        verbosity=0,
        categorical_threshold=10
    )

train_df, test_df = train_test_split(df, test_size=0.2, random_state=42)

# Fit the optimizer
optimizer.fit(train_df)

# Get the best model parameters and print them
results = optimizer.best_model_params(test_df)
print(results)
