Exemplo n.º 1
0
def find_best_svc(**params):
    parameters = {
        'C': [0.01, 0.1, 1, 10, 100, 1000],
        'kernel': ['poly', 'rbf', 'sigmoid'],
        'gamma': [0.0001, 0.001, 0.01, 0.1, 0.5],
    }
    return GridSearchCV(SVC(**params), parameters)