Случайный поиск продолжает работать - PullRequest
0 голосов
/ 24 февраля 2020

Я пытаюсь запустить случайный поиск по гиперпараметру для алгоритма SV C, в частности, для параметра C и гаммы. Однако, как только я нажал кнопку «Выполнить», я получил предупреждение от пользователя, и процесс продолжался бесконечно. Я поставил код Python ниже, и кто-нибудь знает, где я ошибся? Спасибо!

UserWarning: наименее заполненный класс в y имеет только 1 член, что меньше, чем n_splits = 3. % (min_groups, self.n_splits)), предупреждение пользователя)

`

from sklearn.model_selection import RandomizedSearchCV
from scipy.stats import reciprocal, uniform
from sklearn.svm import SVC
param_distributions = {"gamma": reciprocal(0.01, 0.1),
                       "C": uniform(300, 500)}
random_search_svc = RandomizedSearchCV(SVC(kernel='rbf'), param_distributions, cv=3, n_iter=10, random_state=42)
random_search_svc.fit(X_tr, np.ravel(y_tr))

`

...