40
правок
Изменения
→Пример использования с помощью библиотеки xgboost
==Пример использования с помощью библиотеки xgboost==
Загрузка датасета.
'''from ''' sklearn '''import ''' datasets iris = datasets.'''load_iris'''() X = iris.'''data''' y = iris.'''target'''
Разделение датасета на обучающую/тестовую выборку.
'''from ''' sklearn.cross_validation '''import ''' train_test_split X_train, X_test, y_train, y_test = '''train_test_split'''(X, y, test_size=0.2, random_state=42)
Импорт ''XGBoost'' и создание необходимых объектов.
'''import ''' xgboost as xgb dtrain = xgb.'''DMatrix'''(X_train, label=y_train) dtest = xgb.'''DMatrix'''(X_test, label=y_test)
Задание параметров модели.
Обучение.
bst = xgb.'''train'''(param, dtrain, num_round) preds = bst.'''predict'''(dtest)
Определение качества модели на тестовой выборке.
'''import ''' numpy '''as ''' np '''from ''' sklearn.metrics '''import ''' precision_score
best_preds = np.asarray([np.argmax(line) for line in preds])
'''print ''' precision_score(y_test, best_preds, average='macro')
==См. также==
