Skip to content

Commit 69ec309

Browse files
Update lgbm_regressor.py
1 parent e9425d7 commit 69ec309

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

machine_learning/lgbm_regressor.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -17,10 +17,10 @@ def data_handling(data: dict) -> tuple:
1717
return (data["data"], data["target"])
1818

1919

20-
def lgbm_regressor(features: np.ndarray, target: np.ndarray,
20+
def lgbm_regressor(features: np.ndarray, target: np.ndarray,
2121
test_features: np.ndarray) -> np.ndarray:
2222
"""
23-
>>> lgbm_regressor(np.array([[0.12, 0.02, 0.01, 0.25, 0.09]]),
23+
>>> lgbm_regressor(np.array([[0.12, 0.02, 0.01, 0.25, 0.09]]),
2424
... np.array([1]), np.array([[0.11, 0.03, 0.02, 0.28, 0.08]]))
2525
array([[0.98]], dtype=float32)
2626
"""

0 commit comments

Comments
 (0)