Skip to content

XGBoost #5

@bdiptesh

Description

@bdiptesh

Is your feature request related to a problem? Please describe.
XGBoost module

Describe the solution you'd like
XGBoost module for both classification and regression. Determine optimal hyperparameters

Methods:

  1. Classification
  2. Regression

Expected input

df: pandas.DataFrame
y_var: List[str]
x_var: List[str]
method: Union[str]
param: Dict

Expected API

fit
predict

Acceptance criteria

  • Integration tests

Tasks

  • Define integration tests
  • First pass implementation of XGBoost
  • Modular implementation

Metadata

Metadata

Labels

featureNew feature or requeststatsStatisticstestsIntegration/Unit tests

Projects

Status

Done

Milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions