/
githubmirror
/
scikit-learn
Обзор
Документация
Войти
/
githubmirror
/
scikit-learn
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
main
asv_benchmarks/benchmarks/svm.py
30 строк
762 B
Thomas J. Fan
MNT Applies black formatting to most of the code base (#18948)
17 июн 2021, 21:21
Не верифицирован
17 июн 2021, 21:21
82df489
Код
Авторство
О чём код?
from sklearn.svm import SVC from .common import Benchmark, Estimator, Predictor from .datasets import _synth_classification_dataset from .utils import make_gen_classif_scorers class SVCBenchmark(Predictor, Estimator, Benchmark): """Benchmarks for SVC.""" param_names = ["kernel"] params = (["linear", "poly", "rbf", "sigmoid"],) def setup_cache(self): super().setup_cache() def make_data(self, params): return _synth_classification_dataset() def make_estimator(self, params): (kernel,) = params estimator = SVC( max_iter=100, tol=1e-16, kernel=kernel, random_state=0, gamma="scale" ) return estimator def make_scorers(self): make_gen_classif_scorers(self)