Class TGBMClassifier
Synopsis
class TGBMClassifier(TGBMModel, ThundergbmClassifierBase)
Description
No description yet.
Mentioned in
- Getting Started / Quick Install
Inheritance
Ancestors: ThundergbmClassifierBase, TGBMModel
Methods
__init__ |
Source
Lines 349-364 in python/thundergbm/thundergbm.py.
class TGBMClassifier(TGBMModel, ThundergbmClassifierBase):
_impl = 'classifier'
def __init__(self, depth=6, n_trees=40,
n_gpus=1, min_child_weight=1.0, lambda_tgbm=1.0, gamma=1.0, max_num_bin=255,
verbose=1, column_sampling_rate=1.0, bagging=0,
n_parallel_trees=1, learning_rate=1.0, objective="multi:softmax",
num_class=2, tree_method="auto"):
super().__init__(depth=depth, n_trees=n_trees,
n_gpus=n_gpus, min_child_weight=min_child_weight, lambda_tgbm=lambda_tgbm, gamma=gamma,
max_num_bin=max_num_bin,
verbose=verbose, column_sampling_rate=column_sampling_rate, bagging=bagging,
n_parallel_trees=n_parallel_trees, learning_rate=learning_rate, objective=objective,
num_class=num_class, tree_method=tree_method)
class TGBMRegressor(TGBMModel, ThundergbmRegressorBase):