Skip to content

Commit 3e93a6c

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

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

machine_learning/lgbm_regressor.py

+2-4
Original file line numberDiff line numberDiff line change
@@ -25,9 +25,7 @@ def lgbm_regressor(
2525
... np.array([[0.11, 0.03, 0.02, 0.28, 0.08]]))
2626
array([[0.98]], dtype=float32)
2727
"""
28-
lgbm = LGBMRegressor(
29-
verbosity=0, random_state=42
30-
)
28+
lgbm = LGBMRegressor(verbosity=0, random_state=42)
3129
lgbm.fit(features, target)
3230
# Predict target for test data
3331
predictions = lgbm.predict(test_features)
@@ -46,7 +44,7 @@ def main() -> None:
4644
Mean Square Error: 0.15 (approx.)
4745
"""
4846
# Load Bank Marketing dataset
49-
bank_data = fetch_openml(name='bank-marketing', version=1, as_frame=False)
47+
bank_data = fetch_openml(name="bank-marketing", version=1, as_frame=False)
5048
data, target = data_handling(bank_data)
5149
x_train, x_test, y_train, y_test = train_test_split(
5250
data, target, test_size=0.25, random_state=1

0 commit comments

Comments
 (0)