Skip to content

Commit c78ede9

Browse files
authored
1 parent 6496c2c commit c78ede9

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

machine_learning/xgboostclassifier.py

+2-3
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88

99

1010
def data_handling(data: dict) -> tuple:
11-
# Split dataset into train and test data
11+
# Split dataset into features and target
1212
# data is features
1313
"""
1414
>>> data_handling(({'data':'[5.1, 3.5, 1.4, 0.2]','target':([0])}))
@@ -33,8 +33,7 @@ def xgboost(features: np.ndarray, target: np.ndarray) -> XGBClassifier:
3333
max_delta_step=0, max_depth=6, max_leaves=0, min_child_weight=1,
3434
missing=nan, monotone_constraints='()', n_estimators=100,
3535
n_jobs=0, num_parallel_tree=1, predictor='auto', random_state=0,
36-
reg_alpha=0, reg_lambda=1,scale_pos_weight: 1, seed: None,
37-
silent: None, subsample: 1, verbosity: 1)
36+
reg_alpha=0, reg_lambda=1, ...)
3837
"""
3938
classifier = XGBClassifier()
4039
classifier.fit(features, target)

0 commit comments

Comments
 (0)