ml-lib/CodeLib

[Feature]: XGBoost

bdiptesh opened this issue · 0 comments

bdiptesh commented

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(s)

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

Expected output(s)

fit
predict

Additional context

No response

Acceptance criteria

  • Integration tests

Version

v0.4.0