Skip to content

Instantly share code, notes, and snippets.

@JaeDukSeo
Created June 14, 2018 00:08
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save JaeDukSeo/ceb627b6b8b3ca9cd2c02f6f2c58db93 to your computer and use it in GitHub Desktop.
Save JaeDukSeo/ceb627b6b8b3ca9cd2c02f6f2c58db93 to your computer and use it in GitHub Desktop.
from sklearn import datasets
from sklearn.model_selection import train_test_split
from sklearn.model_selection import cross_val_score
X, y = datasets.make_classification(n_samples=10000, n_features=20,
n_informative=2, n_redundant=10,
random_state=42)
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.3,
random_state=42)
from xgboost.sklearn import XGBClassifier
from xgboost.sklearn import XGBRegressor
xclas = XGBClassifier() # and for classifier
xclas.fit(X_train, y_train)
xclas.predict(X_test)
print(cross_val_score(xclas, X_train, y_train) )
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment