We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 1f3affb commit 307122fCopy full SHA for 307122f
machine_learning/xgboostclassifier.py
@@ -23,7 +23,7 @@ def data_handling(data: dict) -> tuple:
23
24
def xgboost(features: np.ndarray, target: np.ndarray) -> XGBClassifier:
25
"""
26
- >>> xgboost(np.array([[5.1, 3.6, 1.4, 0.2],[4.6, 3.4, 1.4, 0.7]]), np.array([1,2]))
+ >>> xgboost(np.array([[5.1, 3.6, 1.4, 0.2],[4.6, 3.4, 1.4, 0.7]]), np.array([0,1]))
27
XGBClassifier()
28
29
classifier = XGBClassifier()
0 commit comments