사이킷런에서 양성 클래스는 레이블이 1인 클래스입니다. 양성 레이블을 바꾸고 싶다면 make_scorer 함수를 사용하여 자신만의 함수를 만들 수 있습니다. 그다음 GridSearchCVscoring 매개변수에 전달할 수 있습니다. 예를 들어 f1_score를 측정 지표로 사용하는 경우는 다음과 같습니다.

    >>> from sklearn.metrics import make_scorer, f1_score
    >>> scorer = make_scorer(f1_score, pos_label=0)
    >>> c_gamma_range = [0.01, 0.1, 1.0, 10.0]
    >>> param_grid = [{'svc__C': c_gamma_range,
    ...                'svc__kernel': ['linear']},
    ...               {'svc__C': c_gamma_range,
    ...                'svc__gamma': c_gamma_range,
    ...                'svc__kernel': ['rbf']}]
    >>> gs = GridSearchCV(estimator=pipe_svc,
    ...                   param_grid=param_grid,
    ...                   scoring=scorer,
    ...                   cv=10)
    >>> gs = gs.fit(X_train, y_train)
    >>> print(gs.best_score_)
    0.9861994953378878
    >>> print(gs.best_params_)
    {'svc__C': 10.0, 'svc__gamma': 0.01, 'svc__kernel': 'rbf'}
    신간 소식 구독하기
    뉴스레터에 가입하시고 이메일로 신간 소식을 받아 보세요.