Skip to content

Commit 307122f

Browse files
authored
1 parent 1f3affb commit 307122f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

machine_learning/xgboostclassifier.py

+1-1
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([1,2]))
26+
>>> xgboost(np.array([[5.1, 3.6, 1.4, 0.2],[4.6, 3.4, 1.4, 0.7]]), np.array([0,1]))
2727
XGBClassifier()
2828
"""
2929
classifier = XGBClassifier()

0 commit comments

Comments
 (0)