-
Notifications
You must be signed in to change notification settings - Fork 1
Closed
Labels
featureNew feature or requestNew feature or requeststatsStatisticsStatisticstestsIntegration/Unit testsIntegration/Unit tests
Milestone
Description
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:
- Classification
- Regression
Expected input
df: pandas.DataFrame
y_var: List[str]
x_var: List[str]
method: Union[str]
param: DictExpected API
fit
predictAcceptance criteria
- Integration tests
Tasks
- Define integration tests
- First pass implementation of XGBoost
- Modular implementation
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
featureNew feature or requestNew feature or requeststatsStatisticsStatisticstestsIntegration/Unit testsIntegration/Unit tests
Projects
Status
Done