Class TGBMClassifier

Synopsis

class TGBMClassifier(TGBMModel, ThundergbmClassifierBase)

Description

No description yet.

Mentioned in

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):





Add Discussion as Guest

Log in