Metadata-Version: 2.1
Name: sklearn-cv-pandas
Version: 0.0.2
Summary: RandomizedSearchCV/GridSearchCV with pandas.DataFrame interface
Home-page: https://github.com/not-so-fat/sklearn_cv_pandas
Author: @not-so-fat
License: UNKNOWN
Platform: UNKNOWN
Classifier: Programming Language :: Python :: 3
Classifier: License :: OSI Approved :: MIT License
Classifier: Operating System :: OS Independent
Requires-Python: >=3.6
Description-Content-Type: text/markdown
Requires-Dist: pandas
Requires-Dist: numpy
Requires-Dist: sklearn

# sklearn-cv-pandas
RandomizedSearchCV/GridSearchCV with pandas.DataFrame interface

## Installation
```
pip install sklearn_cv_pandas
```

## Usage

To tune hyper parameters, instantiate CV as the same as original ones, and use methods `fit_sv_pandas` or `fit_cv_pandas`
```
from scipy import stats
from sklearn import linear_model
from sklearn_cv_pandas import RandomizedSearchCV

estimator = linear_model.Lasso()
param_dist = dict(alpha=stats.loguniform(1e-5, 10))
cv = RandomizedSearchCV(estimator, param_dist, scoring="mean_absolute_error")
model = cv.fit_cv_pandas(
    df, target_column="y", feature_columns=["x{}".format(i) for i in range(100)], n_fold=5
)
```

To make prediction, use method `predict` of the output of `fit_sv_pandas` or `fit_cv_pandas`

```
model.predict(df)
```

