Skip to content

Commit e9425d7

Browse files
Update lgbm_classifier.py
1 parent e9dc1aa commit e9425d7

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
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
)

0 commit comments

Comments
 (0)