mlforgex 1.0.0__tar.gz → 1.0.2__tar.gz

mlforgex-1.0.2/PKG-INFO

@@ -0,0 +1,142 @@
+Metadata-Version: 2.4
+Name: mlforgex
+Version: 1.0.2
+Summary: Lightweight ML utility for automated training, evaluation, and prediction with CLI and Python API support
+Home-page: https://github.com/yourusername/mlforge
+Author: Priyanshu Mathur
+Author-email: mathurpriyanshu2006@gmail.com
+License: MIT
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Operating System :: OS Independent
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: pandas
+Requires-Dist: numpy
+Requires-Dist: seaborn
+Requires-Dist: matplotlib
+Requires-Dist: scikit-learn
+Requires-Dist: xgboost
+Requires-Dist: imbalanced-learn
+Dynamic: author
+Dynamic: author-email
+Dynamic: classifier
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: home-page
+Dynamic: license
+Dynamic: license-file
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary
+
+# mlforgex
+
+**mlforgex** is a Python package that enables easy training, evaluation, and prediction for machine learning models on cleaned dataset. It supports both classification and regression problems, automates preprocessing, model selection, hyperparameter tuning, and generates useful artifacts and plots for analysis.
+
+## Features
+
+- Automatic data preprocessing (missing value handling, encoding, scaling)
+- Imbalance handling (under-sampling, over-sampling)
+- Model selection and evaluation (classification & regression)
+- Hyperparameter tuning with RandomizedSearchCV
+- Artifact saving (model, preprocessor, encoder)
+- Visualization of metrics and learning curves
+- Simple CLI for training and prediction
+
+## Installation
+
+Install mlforge using pip:
+
+```sh
+pip install mlforgex
+```
+stall .
+```
+
+## Requirements
+
+- Python >= 3.8
+- pandas
+- numpy
+- scikit-learn
+- seaborn
+- matplotlib
+- xgboost
+- imbalanced-learn
+
+See [requirements.txt](requirements.txt) for details.
+
+## Usage
+
+### Train a Model
+
+You can train a model using the CLI:
+
+```sh
+mlforge-train --data_path path/to/your/data.csv --dependent_feature TargetColumn --rmse_prob 0.3 --f1_prob 0.7 --n_jobs -1 --n_iter 100 --cv 3
+```
+
+Or programmatically:
+
+```python
+from mlforge import train_model
+
+result = train_model(
+    data_path=<data_path>,
+    dependent_feature=<dependent_feature>,
+    rmse_prob=<rmse_probability>,
+    f1_prob=<f1_probability>,
+    n_jobs=<n_jobs>
+)
+print(result)
+```
+
+### Predict
+
+Use the CLI:
+
+```sh
+mlforge-predict --model_path path/to/model.pkl --preprocessor_path path/to/preprocessor.pkl --input_data path/to/input.csv --encoder_path path/to/encoder.pkl
+```
+
+Or programmatically:
+
+```python
+from mlforge import predict
+
+result = predict(
+    <model.pkl>,
+    <preprocessor.pkl>,
+    <input_data.csv>,
+    <encoder.pkl>
+)
+print(result)
+```
+
+## Artifacts
+
+After training, the following files are saved :
+
+- `model.pkl`: Trained model
+- `preprocessor.pkl`: Preprocessing pipeline
+- `encoder.pkl`: Label encoder (for classification)
+- `Plots/`: Visualizations (correlation heatmap, confusion matrix, ROC curve, etc.)
+
+## Testing
+
+Run tests using pytest:
+
+```sh
+pytest test/
+```
+## Author
+
+Priyanshu Mathur  
+[Portfolio](https://my-portfolio-phi-two-53.vercel.app/)  
+Email: mathurpriyanshu2006@gmail.com
+
+## Project Links
+
+- [PyPI](https://pypi.org/project/mlforgex/)

mlforgex-1.0.2/README.md

@@ -0,0 +1,109 @@
+# mlforgex
+
+**mlforgex** is a Python package that enables easy training, evaluation, and prediction for machine learning models on cleaned dataset. It supports both classification and regression problems, automates preprocessing, model selection, hyperparameter tuning, and generates useful artifacts and plots for analysis.
+
+## Features
+
+- Automatic data preprocessing (missing value handling, encoding, scaling)
+- Imbalance handling (under-sampling, over-sampling)
+- Model selection and evaluation (classification & regression)
+- Hyperparameter tuning with RandomizedSearchCV
+- Artifact saving (model, preprocessor, encoder)
+- Visualization of metrics and learning curves
+- Simple CLI for training and prediction
+
+## Installation
+
+Install mlforge using pip:
+
+```sh
+pip install mlforgex
+```
+stall .
+```
+
+## Requirements
+
+- Python >= 3.8
+- pandas
+- numpy
+- scikit-learn
+- seaborn
+- matplotlib
+- xgboost
+- imbalanced-learn
+
+See [requirements.txt](requirements.txt) for details.
+
+## Usage
+
+### Train a Model
+
+You can train a model using the CLI:
+
+```sh
+mlforge-train --data_path path/to/your/data.csv --dependent_feature TargetColumn --rmse_prob 0.3 --f1_prob 0.7 --n_jobs -1 --n_iter 100 --cv 3
+```
+
+Or programmatically:
+
+```python
+from mlforge import train_model
+
+result = train_model(
+    data_path=<data_path>,
+    dependent_feature=<dependent_feature>,
+    rmse_prob=<rmse_probability>,
+    f1_prob=<f1_probability>,
+    n_jobs=<n_jobs>
+)
+print(result)
+```
+
+### Predict
+
+Use the CLI:
+
+```sh
+mlforge-predict --model_path path/to/model.pkl --preprocessor_path path/to/preprocessor.pkl --input_data path/to/input.csv --encoder_path path/to/encoder.pkl
+```
+
+Or programmatically:
+
+```python
+from mlforge import predict
+
+result = predict(
+    <model.pkl>,
+    <preprocessor.pkl>,
+    <input_data.csv>,
+    <encoder.pkl>
+)
+print(result)
+```
+
+## Artifacts
+
+After training, the following files are saved :
+
+- `model.pkl`: Trained model
+- `preprocessor.pkl`: Preprocessing pipeline
+- `encoder.pkl`: Label encoder (for classification)
+- `Plots/`: Visualizations (correlation heatmap, confusion matrix, ROC curve, etc.)
+
+## Testing
+
+Run tests using pytest:
+
+```sh
+pytest test/
+```
+## Author
+
+Priyanshu Mathur  
+[Portfolio](https://my-portfolio-phi-two-53.vercel.app/)  
+Email: mathurpriyanshu2006@gmail.com
+
+## Project Links
+
+- [PyPI](https://pypi.org/project/mlforgex/)

{mlforgex-1.0.0 → mlforgex-1.0.2}/mlforge/predict.py

@@ -14,10 +14,10 @@ def predict(model_path,preprocessor_path,input_data, encoder_path=None):
 def main():
     import argparse
     parser = argparse.ArgumentParser()
-    parser.add_argument("--model", required=True)
-    parser.add_argument("--input", required=True)
-    parser.add_argument("--preprocessor", required=True)
-    parser.add_argument("--encoder", required=False)
+    parser.add_argument("--model_path", required=True)
+    parser.add_argument("--input_data", required=True)
+    parser.add_argument("--preprocessor_path", required=True)
+    parser.add_argument("--encoder_path", required=False)
     args = parser.parse_args()
-    print(predict(args.model, args.preprocessor, args.input, args.encoder))
+    print(predict(args.model_path, args.preprocessor_path, args.input_data, args.encoder_path))
 

{mlforgex-1.0.0 → mlforgex-1.0.2}/mlforge/train.py

@@ -25,9 +25,12 @@ from sklearn.inspection import permutation_importance
 from sklearn.model_selection import learning_curve
 import warnings
 warnings.filterwarnings("ignore")
-artifacts_path=os.path.join(os.path.dirname(__file__),"artifacts")
-plot_path=os.path.join(artifacts_path,"Plots")
-def train_model(data_path, dependent_feature,rmse_prob,f1_prob,n_jobs=-1,n_iter=100,cv=3):
+def train_model(data_path, dependent_feature,rmse_prob,f1_prob,n_jobs=-1,n_iter=100,cv=3,artifacts_dir=None):
+    if artifacts_dir:
+        artifacts_path = os.path.join(artifacts_dir, "artifacts")
+    else:
+        artifacts_path = os.path.join(os.getcwd(), "artifacts")
+    plot_path=os.path.join(artifacts_path,"Plots")
     os.makedirs(artifacts_path,exist_ok=True)
     print("Getting data from:",data_path)
     data=pd.read_csv(data_path)
@@ -41,7 +44,7 @@ def train_model(data_path, dependent_feature,rmse_prob,f1_prob,n_jobs=-1,n_iter=
             df[i] = df[i].fillna(df[i].mode()[0])
         else:
             df[i] = df[i].fillna(df[i].median())
-            
+    df.drop_duplicates(inplace=True, ignore_index=True)
     mild=False
     moderate=False
     majority=max(df[dependent_feature].value_counts())
@@ -113,17 +116,18 @@ def train_model(data_path, dependent_feature,rmse_prob,f1_prob,n_jobs=-1,n_iter=
     num_features = [i for i in x_train.columns if x_train[i].dtype != "object" and i != dependent_feature]
     print("Categorical features : ",cat_features)
     print("Numerical features : ",num_features)
-    print("Balancing the dataset...")
-    if mild:
-        from imblearn.under_sampling import RandomUnderSampler
-        undersampler = RandomUnderSampler(random_state=42)
-        x_train, y_train = undersampler.fit_resample(x_train, y_train)
-        print("Mild Imbalance Resolved")
-    if moderate:
-        from imblearn.over_sampling import SMOTETomek
-        smote_tomek = SMOTETomek(random_state=42)
-        x_train, y_train = smote_tomek.fit_resample(x_train, y_train)
-        print("Moderate Imbalace Resolved")
+    if classification:
+        print("Balancing the dataset...")
+        if mild :
+            from imblearn.under_sampling import RandomUnderSampler
+            undersampler = RandomUnderSampler(random_state=42)
+            x_train, y_train = undersampler.fit_resample(x_train, y_train)
+            print("Mild Imbalance Resolved")
+        if moderate:
+            from imblearn.over_sampling import SMOTETomek
+            smote_tomek = SMOTETomek(random_state=42)
+            x_train, y_train = smote_tomek.fit_resample(x_train, y_train)
+            print("Moderate Imbalace Resolved")
 
 
     
@@ -762,7 +766,7 @@ def train_model(data_path, dependent_feature,rmse_prob,f1_prob,n_jobs=-1,n_iter=
             "Dropped_Columns":list(dropcorr)
         }
     }
-        plot_classification_metrics(model,x_train, y_train, x_test, y_test, class_names=['No', 'Yes'])
+        plot_classification_metrics(model,x_train, y_train, x_test, y_test, class_names=['No', 'Yes'],plot_path=plot_path)
     else:
         response={
              "message": "Training completed successfully",
@@ -777,11 +781,16 @@ def train_model(data_path, dependent_feature,rmse_prob,f1_prob,n_jobs=-1,n_iter=
             "Dropped_Columns":list(dropcorr)
             }
         }
-        plot_regression_metrics(model, x_train, y_train, x_test, y_test,feature_names)
+        plot_regression_metrics(model, x_train, y_train, x_test, y_test,feature_names,plot_path=plot_path)
     print(response)
+    print("artifacts_path:", artifacts_path)
+    print("model_path:", model_path)
+    print("preprocessor_path:", preprocessor_path)
+    if encoder_path:
+        print("encoder_path:", encoder_path)
     return {"status": "success", "model": "trained_model.pkl"}
 
-def plot_classification_metrics(model, X_train, y_train, X_test, y_test, class_names=None):
+def plot_classification_metrics(model, X_train, y_train, X_test, y_test, plot_path,class_names=None):
     # Predict
     y_pred = model.predict(X_test)
     y_proba = model.predict_proba(X_test)[:, 1] if hasattr(model, "predict_proba") else None
@@ -826,7 +835,7 @@ def plot_classification_metrics(model, X_train, y_train, X_test, y_test, class_n
     ax[1].set_title("Testing Class Distribution")
     plt.savefig(os.path.join(plot_path,"class_distribution.png"), bbox_inches='tight')
     plt.close()
-def plot_regression_metrics(model, X_train, y_train, X_test, y_test,feature_names):
+def plot_regression_metrics(model, X_train, y_train, X_test, y_test,feature_names,plot_path):
     y_pred = model.predict(X_test)
     residuals = y_test - y_pred
 
@@ -867,13 +876,13 @@ def plot_regression_metrics(model, X_train, y_train, X_test, y_test,feature_name
 def main():
     import argparse
     parser = argparse.ArgumentParser()
-    parser.add_argument("--data", required=True)
-    parser.add_argument("--target", required=True)
-    parser.add_argument("--rmse", required=True)
-    parser.add_argument("--f1", required=True)
+    parser.add_argument("--data_path", required=True)
+    parser.add_argument("--dependent_feature", required=True)
+    parser.add_argument("--rmse_prob", required=True)
+    parser.add_argument("--f1_prob", required=True)
     parser.add_argument("--n_jobs", default=-1, type=int)
     parser.add_argument("--n_iter", default=100, type=int)
     parser.add_argument("--cv", default=3, type=int)
     args = parser.parse_args()
-    print(train_model(args.data, args.target, rmse_prob=args.rmse, f1_prob=args.f1, n_jobs=args.n_jobs, n_iter=args.n_iter, cv=args.cv))
+    print(train_model(args.data_path, args.dependent_feature, rmse_prob=args.rmse_prob, f1_prob=args.f1_prob, n_jobs=args.n_jobs, n_iter=args.n_iter, cv=args.cv))
 

mlforgex-1.0.2/mlforgex.egg-info/PKG-INFO

@@ -0,0 +1,142 @@
+Metadata-Version: 2.4
+Name: mlforgex
+Version: 1.0.2
+Summary: Lightweight ML utility for automated training, evaluation, and prediction with CLI and Python API support
+Home-page: https://github.com/yourusername/mlforge
+Author: Priyanshu Mathur
+Author-email: mathurpriyanshu2006@gmail.com
+License: MIT
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Operating System :: OS Independent
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: pandas
+Requires-Dist: numpy
+Requires-Dist: seaborn
+Requires-Dist: matplotlib
+Requires-Dist: scikit-learn
+Requires-Dist: xgboost
+Requires-Dist: imbalanced-learn
+Dynamic: author
+Dynamic: author-email
+Dynamic: classifier
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: home-page
+Dynamic: license
+Dynamic: license-file
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary
+
+# mlforgex
+
+**mlforgex** is a Python package that enables easy training, evaluation, and prediction for machine learning models on cleaned dataset. It supports both classification and regression problems, automates preprocessing, model selection, hyperparameter tuning, and generates useful artifacts and plots for analysis.
+
+## Features
+
+- Automatic data preprocessing (missing value handling, encoding, scaling)
+- Imbalance handling (under-sampling, over-sampling)
+- Model selection and evaluation (classification & regression)
+- Hyperparameter tuning with RandomizedSearchCV
+- Artifact saving (model, preprocessor, encoder)
+- Visualization of metrics and learning curves
+- Simple CLI for training and prediction
+
+## Installation
+
+Install mlforge using pip:
+
+```sh
+pip install mlforgex
+```
+stall .
+```
+
+## Requirements
+
+- Python >= 3.8
+- pandas
+- numpy
+- scikit-learn
+- seaborn
+- matplotlib
+- xgboost
+- imbalanced-learn
+
+See [requirements.txt](requirements.txt) for details.
+
+## Usage
+
+### Train a Model
+
+You can train a model using the CLI:
+
+```sh
+mlforge-train --data_path path/to/your/data.csv --dependent_feature TargetColumn --rmse_prob 0.3 --f1_prob 0.7 --n_jobs -1 --n_iter 100 --cv 3
+```
+
+Or programmatically:
+
+```python
+from mlforge import train_model
+
+result = train_model(
+    data_path=<data_path>,
+    dependent_feature=<dependent_feature>,
+    rmse_prob=<rmse_probability>,
+    f1_prob=<f1_probability>,
+    n_jobs=<n_jobs>
+)
+print(result)
+```
+
+### Predict
+
+Use the CLI:
+
+```sh
+mlforge-predict --model_path path/to/model.pkl --preprocessor_path path/to/preprocessor.pkl --input_data path/to/input.csv --encoder_path path/to/encoder.pkl
+```
+
+Or programmatically:
+
+```python
+from mlforge import predict
+
+result = predict(
+    <model.pkl>,
+    <preprocessor.pkl>,
+    <input_data.csv>,
+    <encoder.pkl>
+)
+print(result)
+```
+
+## Artifacts
+
+After training, the following files are saved :
+
+- `model.pkl`: Trained model
+- `preprocessor.pkl`: Preprocessing pipeline
+- `encoder.pkl`: Label encoder (for classification)
+- `Plots/`: Visualizations (correlation heatmap, confusion matrix, ROC curve, etc.)
+
+## Testing
+
+Run tests using pytest:
+
+```sh
+pytest test/
+```
+## Author
+
+Priyanshu Mathur  
+[Portfolio](https://my-portfolio-phi-two-53.vercel.app/)  
+Email: mathurpriyanshu2006@gmail.com
+
+## Project Links
+
+- [PyPI](https://pypi.org/project/mlforgex/)

mlforgex-1.0.2/mlforgex.egg-info/requires.txt

@@ -0,0 +1,7 @@
+pandas
+numpy
+seaborn
+matplotlib
+scikit-learn
+xgboost
+imbalanced-learn

mlforgex-1.0.2/setup.py

@@ -0,0 +1,39 @@
+from setuptools import setup, find_packages
+from pathlib import Path
+
+this_directory = Path(__file__).parent
+long_description = (this_directory / "README.md").read_text()
+setup(
+    name="mlforgex",
+    version="1.0.2",
+    packages=find_packages(),
+    install_requires = [
+    "pandas",
+    "numpy",
+    "seaborn",
+    "matplotlib",
+    "scikit-learn",
+    "xgboost",
+    "imbalanced-learn"
+]
+,
+    long_description=long_description,
+    long_description_content_type="text/markdown",
+    entry_points={
+        "console_scripts": [
+            "mlforge-train=mlforge.train:main",
+            "mlforge-predict=mlforge.predict:main"
+        ]
+    },
+    author="Priyanshu Mathur",
+    author_email="mathurpriyanshu2006@gmail.com",
+    description="Lightweight ML utility for automated training, evaluation, and prediction with CLI and Python API support",
+    url="https://github.com/yourusername/mlforge",
+    license="MIT",
+    classifiers=[
+         "License :: OSI Approved :: MIT License",
+        "Programming Language :: Python :: 3",
+        "Operating System :: OS Independent",
+    ],
+    python_requires=">=3.8",
+)

mlforgex-1.0.0/PKG-INFO

@@ -1,22 +0,0 @@
-Metadata-Version: 2.4
-Name: mlforgex
-Version: 1.0.0
-Summary: MLForge: Train and evaluate ML models easily
-Home-page: https://github.com/yourusername/mlforge
-Author: Priyanshu Mathur
-Author-email: mathurpriyanshu2006@gmail.com
-Classifier: Programming Language :: Python :: 3
-Classifier: Operating System :: OS Independent
-Requires-Python: >=3.8
-License-File: LICENSE
-Requires-Dist: pandas
-Requires-Dist: numpy
-Requires-Dist: scikit-learn
-Dynamic: author
-Dynamic: author-email
-Dynamic: classifier
-Dynamic: home-page
-Dynamic: license-file
-Dynamic: requires-dist
-Dynamic: requires-python
-Dynamic: summary

mlforgex-1.0.0/README.md

@@ -1,121 +0,0 @@
-# MLForge
-
-**MLForge** is a Python package that enables easy training, evaluation, and prediction for machine learning models. It supports both classification and regression problems, automates preprocessing, model selection, hyperparameter tuning, and generates useful artifacts and plots for analysis.
-
-## Features
-
-- Automatic data preprocessing (missing value handling, encoding, scaling)
-- Imbalance handling (under-sampling, over-sampling)
-- Model selection and evaluation (classification & regression)
-- Hyperparameter tuning with RandomizedSearchCV
-- Artifact saving (model, preprocessor, encoder)
-- Visualization of metrics and learning curves
-- Simple CLI for training and prediction
-
-## Installation
-
-Install MLForge using pip:
-
-```sh
-pip install mlforge
-```
-
-Or clone the repository and install locally:
-
-```sh
-git clone https://github.com/yourusername/mlforge.git
-cd mlforge
-pip install .
-```
-
-## Requirements
-
-- Python >= 3.8
-- pandas
-- numpy
-- scikit-learn
-- seaborn
-- matplotlib
-- xgboost
-- imbalanced-learn
-
-See [requirements.txt](requirements.txt) for details.
-
-## Usage
-
-### Train a Model
-
-You can train a model using the CLI:
-
-```sh
-mlforge-train --data mlforge/diabetes_cleaned.csv --target Outcome --rmse 0.3 --f1 0.7
-```
-
-Or programmatically:
-
-```python
-from mlforge import train_model
-
-result = train_model(
-    "mlforge/diabetes_cleaned.csv",
-    "Outcome",
-    rmse_prob=0.3,
-    f1_prob=0.7,
-    n_jobs=-1
-)
-print(result)
-```
-
-### Predict
-
-Use the CLI:
-
-```sh
-mlforge-predict --model mlforge/artifacts/model.pkl --preprocessor mlforge/artifacts/preprocessor.pkl --input mlforge/input.csv --encoder mlforge/artifacts/encoder.pkl
-```
-
-Or programmatically:
-
-```python
-from mlforge import predict
-
-result = predict(
-    "mlforge/artifacts/model.pkl",
-    "mlforge/artifacts/preprocessor.pkl",
-    "mlforge/input.csv",
-    "mlforge/artifacts/encoder.pkl"
-)
-print(result)
-```
-
-## Artifacts
-
-After training, the following files are saved in `mlforge/artifacts/`:
-
-- `model.pkl`: Trained model
-- `preprocessor.pkl`: Preprocessing pipeline
-- `encoder.pkl`: Label encoder (for classification)
-- `Plots/`: Visualizations (correlation heatmap, confusion matrix, ROC curve, etc.)
-
-## Testing
-
-Run tests using pytest:
-
-```sh
-pytest test/
-```
-
-## License
-
-[MIT License](https://github.com/dhgefergfefruiwefhjhcduc/ML_Forge?tab=MIT-1-ov-file)
-
-## Author
-
-Priyanshu Mathur  
-[Portfolio](https://my-portfolio-phi-two-53.vercel.app/)  
-Email: mathurpriyanshu2006@gmail.com
-
-## Project Links
-
-- [PyPI](https://pypi.org/project/mlforge/)
-- [GitHub](https://github.com/yourusername/mlforge)

mlforgex-1.0.0/mlforgex.egg-info/PKG-INFO

@@ -1,22 +0,0 @@
-Metadata-Version: 2.4
-Name: mlforgex
-Version: 1.0.0
-Summary: MLForge: Train and evaluate ML models easily
-Home-page: https://github.com/yourusername/mlforge
-Author: Priyanshu Mathur
-Author-email: mathurpriyanshu2006@gmail.com
-Classifier: Programming Language :: Python :: 3
-Classifier: Operating System :: OS Independent
-Requires-Python: >=3.8
-License-File: LICENSE
-Requires-Dist: pandas
-Requires-Dist: numpy
-Requires-Dist: scikit-learn
-Dynamic: author
-Dynamic: author-email
-Dynamic: classifier
-Dynamic: home-page
-Dynamic: license-file
-Dynamic: requires-dist
-Dynamic: requires-python
-Dynamic: summary

mlforgex-1.0.0/mlforgex.egg-info/requires.txt

@@ -1,3 +0,0 @@
-pandas
-numpy
-scikit-learn

mlforgex-1.0.0/setup.py

@@ -1,27 +0,0 @@
-from setuptools import setup, find_packages
-
-setup(
-    name="mlforgex",
-    version="1.0.0",
-    packages=find_packages(),
-    install_requires=[
-        "pandas",
-        "numpy",
-        "scikit-learn"
-    ],
-    entry_points={
-        "console_scripts": [
-            "mlforge-train=mlforge.train:main",
-            "mlforge-predict=mlforge.predict:main"
-        ]
-    },
-    author="Priyanshu Mathur",
-    author_email="mathurpriyanshu2006@gmail.com",
-    description="MLForge: Train and evaluate ML models easily",
-    url="https://github.com/yourusername/mlforge",
-    classifiers=[
-        "Programming Language :: Python :: 3",
-        "Operating System :: OS Independent",
-    ],
-    python_requires=">=3.8",
-)