Defines search spaces for scikit-lean estimators
Project description
sksearchspace
Scikit-learn Search Space Configurations with curated search spaces for scikit-learn estimators.
Usage
from sksearchspace import SearchSpace
from sklearn.tree import DecisionTreeClassifier
estimator_space = SearchSpace.for_sklearn_estimator(DecisionTreeClassifier, seed=42)
estimator_space.sample()
# {'criterion': 'entropy','min_samples_leaf': 15, 'min_samples_split': 11}
estimator_space.sample()
# {'criterion': 'entropy', 'min_samples_leaf': 12, 'min_samples_split': 4}
sksearchspace
uses ConfigSpace for sampling. The ConfigSpace
configuration can be accessed through an attribute:
estimator_space.configuration
# Configuration space object:
# Hyperparameters:
# criterion, Type: Categorical, Choices: {gini, entropy}, Default: gini
# min_samples_leaf, Type: UniformInteger, Range: [1, 20], Default: 1
# min_samples_split, Type: UniformInteger, Range: [2, 20], Default: 2
License
Copyright (c) 2020 Thomas J. Fan
Distributed under the terms of the MIT license, pytest is free and open source software.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Close
Hashes for sksearchspace-2020.7.0.0.23.1.dev0.tar.gz
Algorithm | Hash digest | |
---|---|---|
SHA256 | f69151e8082455e339f7e590dd0b2001c1241dd57a16cc466a33bad04606c6cf |
|
MD5 | 9c6b9b2268cdc6b7655c93388b89966c |
|
BLAKE2b-256 | 2c49bdc1b5e622ad8e9b22d66a9759d3afa959bef76ebfd8c17248b5887227da |
Close
Hashes for sksearchspace-2020.7.0.0.23.1.dev0-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 3cf54d154e53d5c3dca59fcbb85cce05f50f6f2786bc5d39b79b11652e1e9395 |
|
MD5 | 3d7d16c9c2c26524af9e7e9696ec74c9 |
|
BLAKE2b-256 | 410aa592259f63a2da0eceb4ae8cf19c8bc81519a28c5d344cf3d45c64229acc |