Skip to content

Commit 7d42c2e

Browse files
authored
Update xgboostclassifier.py
1 parent 307122f commit 7d42c2e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

machine_learning/xgboostclassifier.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ def data_handling(data: dict) -> tuple:
2323

2424
def xgboost(features: np.ndarray, target: np.ndarray) -> XGBClassifier:
2525
"""
26-
>>> xgboost(np.array([[5.1, 3.6, 1.4, 0.2],[4.6, 3.4, 1.4, 0.7]]), np.array([0,1]))
26+
>>> xgboost(np.array([[5.1, 3.6, 1.4, 0.2]]), np.array([1]))
2727
XGBClassifier()
2828
"""
2929
classifier = XGBClassifier()

0 commit comments

Comments
 (0)