Skip to content

Commit 7298a54

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent f0ceb04 commit 7298a54

File tree

2 files changed

+6
-5
lines changed

2 files changed

+6
-5
lines changed

machine_learning/lgbm_classifier.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,7 @@ def main() -> None:
3535
Bank Marketing Dataset is used to demonstrate the algorithm.
3636
"""
3737
# Load Bank Marketing dataset
38-
bank_data = fetch_openml(name='bank-marketing', version=1, as_frame=False)
38+
bank_data = fetch_openml(name="bank-marketing", version=1, as_frame=False)
3939
data, target = data_handling(bank_data)
4040
x_train, x_test, y_train, y_test = train_test_split(
4141
data, target, test_size=0.25, random_state=1
@@ -48,7 +48,7 @@ def main() -> None:
4848
lgbm_classifier_model,
4949
x_test,
5050
y_test,
51-
display_labels=['No', 'Yes'],
51+
display_labels=["No", "Yes"],
5252
cmap="Blues",
5353
normalize="true",
5454
)

machine_learning/lgbm_regressor.py

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

1919

20-
def lgbm_regressor(features: np.ndarray, target: np.ndarray,
21-
test_features: np.ndarray) -> np.ndarray:
20+
def lgbm_regressor(
21+
features: np.ndarray, target: np.ndarray, test_features: np.ndarray
22+
) -> np.ndarray:
2223
"""
2324
>>> lgbm_regressor(np.array([[0.12, 0.02, 0.01, 0.25, 0.09]]),
2425
... np.array([1]), np.array([[0.11, 0.03, 0.02, 0.28, 0.08]]))
@@ -39,7 +40,7 @@ def main() -> None:
3940
Bank Marketing Dataset is used to demonstrate the algorithm.
4041
"""
4142
# Load Bank Marketing dataset
42-
bank_data = fetch_openml(name='bank-marketing', version=1, as_frame=False)
43+
bank_data = fetch_openml(name="bank-marketing", version=1, as_frame=False)
4344
data, target = data_handling(bank_data)
4445
x_train, x_test, y_train, y_test = train_test_split(
4546
data, target, test_size=0.25, random_state=1

0 commit comments

Comments
 (0)