0
$\begingroup$

I want to compare which technique has higher memory utilization while training on the same dataset

$\endgroup$

1 Answer 1

2
$\begingroup$

Try to use memory_profiler. All what you need to do is to decorate your training function:

@profile
def train_xgb():
   gb = xgb.XGBClassifier(learning_rate=0.1, n_estimators=100, subsample=0.8, max_depth=6)
   gb.fit(new_trainX,new_trainY)
$\endgroup$

Your Answer

By clicking “Post Your Answer”, you agree to our terms of service and acknowledge you have read our privacy policy.

Not the answer you're looking for? Browse other questions tagged or ask your own question.