Skip to content

Instantly share code, notes, and snippets.

@rupak-roy
Created July 11, 2022 05:54
Show Gist options
  • Save rupak-roy/429dd71a35d7b85b1fa752fb8ba30da5 to your computer and use it in GitHub Desktop.
Save rupak-roy/429dd71a35d7b85b1fa752fb8ba30da5 to your computer and use it in GitHub Desktop.
AUTOXGBOOST + Optuna An auto version of one of the most powerful machine learning XGBOOST Pt.2
# Start AutoXGB training
axgb = AutoXGB(
train_filename=train_filename,
output=output,
test_filename=test_filename,
task=task,
targets=targets,
use_gpu=use_gpu,
num_folds=num_folds,
seed=seed,
num_trials=num_trials,
time_limit=time_limit,
fast=fast,
features=features,
categorical_features=categorical_features,
)
axgb.train()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment