ordinalgbt.lgb
Ordinal classifier lightgbm implementation
Classes
Module Contents
- class ordinalgbt.lgb.LGBMOrdinal(boosting_type: str = 'gbdt', num_leaves: int = 31, max_depth: int = -1, learning_rate: float = 0.1, n_estimators: int = 100, subsample_for_bin: int = 200000, objective='immediate-thresholds', class_weight=None, min_split_gain: float = 0.0, min_child_weight: float = 0.001, min_child_samples: int = 20, subsample: float = 1.0, subsample_freq: int = 0, colsample_bytree: float = 1.0, reg_alpha: float = 0.0, reg_lambda: float = 0.0, random_state=None, n_jobs: int = -1, silent='warn', importance_type: str = 'split', **kwargs)
Bases:
lightgbm.LGBMRegressor- _initialise_theta()
- _lgb_loss_factory()
- static _alpha_loss_factory(y_true, y_preds)
Creates loss parametrised by alpha
- _optimise_alpha(y_true, y_preds)
Takes loss parametrised by alpha and optimises it. Can optionally take in gradient.
- _initialise_objective(y)
initialises the objective by creating the loss and setting the class attributes
- _output_to_probability(output)
- _hot_start(X, y, hot_start_iterations=5, **kwargs)
TODO
- fit(X, y, hot_start_iterations=5, sample_weight=None, init_score=None, eval_set=None, eval_names=None, eval_sample_weight=None, eval_init_score=None, eval_metric=None, early_stopping_rounds=None, verbose='warn', feature_name='auto', categorical_feature='auto', callbacks=None, init_model=None) LGBMOrdinal
Docstring is inherited from the LGBMModel.
- _fit(X, y, sample_weight=None, init_score=None, eval_set=None, eval_names=None, eval_sample_weight=None, eval_init_score=None, eval_metric=None, early_stopping_rounds=None, verbose='warn', feature_name='auto', categorical_feature='auto', callbacks=None, init_model=None) LGBMOrdinal
Docstring is inherited from the LGBMModel.
- predict(X, start_iteration=0, num_iteration=None, pred_leaf=False, pred_contrib=False, **kwargs)
- predict_proba(X, raw_score=False, start_iteration=0, num_iteration=None, pred_leaf=False, pred_contrib=False, **kwargs)