Giter Site home page Giter Site logo

Comments (1)

rayeaster avatar rayeaster commented on August 15, 2024 1

working code using sklearn api for: exisiting metric(self-created metric works simliar with native/standard api), evaluation result print, early stop

from xgboost.sklearn import XGBClassifier
from sklearn.datasets import load_svmlight_files
from sklearn.metrics import accuracy_score

#using sklearn api
X_train, y_train, X_test, y_test = load_svmlight_files(('../data/agaricus.txt.train', 'C:/users/lma/Desktop/xgb/data/agaricus.txt.test'))

params = {
    'objective': 'binary:logistic',
    'max_depth': 1,
    'learning_rate': 0.5,
    'silent': 1.0,
    'n_estimators': 1500
}

#evaluate result with specified metric and early stop
eval_set = [(X_test, y_test)]
#you could use self created metric here simply by setting eval_metric=misclassified
bst = XGBClassifier(**params).fit(X_train, y_train, early_stopping_rounds=10, eval_metric="logloss", eval_set=eval_set, verbose=True)
results = bst.evals_result()
print('evaluation result:', results)

y_pred = bst.predict(X_test)
predictions = [round(value) for value in y_pred]
accuracy = accuracy_score(y_test, predictions)
print("Accuracy: %.2f%%" % (accuracy * 100.0))

from docker-course-xgboost.

Related Issues (10)

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.