Tools for doing hyperparameter search with Scikit-Learn and Dask
Project description
Tools for performing hyperparameter search with Scikit-Learn and Dask.
This library provides implementations of Scikit-Learn’s GridSearchCV and RandomizedSearchCV. They implement many (but not all) of the same parameters, and should be a drop-in replacement for the subset that they do implement. For certain problems, these implementations can be more efficient than those in Scikit-Learn, as they can avoid expensive repeated computations.
For more information, check out the documentation.
Install
Dask-searchcv is available via pip:
$ pip install dask-searchcv
Example
from sklearn.datasets import load_digits
from sklearn.svm import SVC
import dask_searchcv as dcv
import numpy as np
digits = load_digits()
param_space = {'C': np.logspace(-4, 4, 9),
'gamma': np.logspace(-4, 4, 9),
'class_weight': [None, 'balanced']}
model = SVC(kernel='rbf')
search = dcv.GridSearchCV(model, param_space, cv=3)
search.fit(digits.data, digits.target)
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
dask-searchcv-0.0.2.tar.gz
(47.2 kB
view hashes)
Built Distribution
Close
Hashes for dask_searchcv-0.0.2-py2.py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 46e6b0918fddbc41fc633769648c3086ea697139822ba59bec0d477c617d63ed |
|
MD5 | 58d732b7ef857c5a87277d03550f862d |
|
BLAKE2b-256 | 0305d404348b6564344a2ad30cc794818f8da51421773a45e5abbe2c99e30789 |