Skip to content

Commit 7ddda2c

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 758ae1d commit 7ddda2c

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

machine_learning/catboost_regressor.py

+5-4
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@
2121
def data_handling() -> tuple:
2222
"""
2323
Loads and handles the California Housing dataset (replacement for deprecated Boston dataset).
24-
24+
2525
Returns:
2626
tuple: A tuple of (features, target), where both are numpy arrays.
2727
@@ -45,11 +45,11 @@ def data_handling() -> tuple:
4545
def catboost_regressor(features: np.ndarray, target: np.ndarray) -> CatBoostRegressor:
4646
"""
4747
Trains a CatBoostRegressor using the provided features and target values.
48-
48+
4949
Args:
5050
features (np.ndarray): The input features for the regression model.
5151
target (np.ndarray): The target values for the regression model.
52-
52+
5353
Returns:
5454
CatBoostRegressor: A trained CatBoost regressor model.
5555
@@ -67,7 +67,7 @@ def catboost_regressor(features: np.ndarray, target: np.ndarray) -> CatBoostRegr
6767
def main() -> None:
6868
"""
6969
Main function to run the CatBoost Regressor example.
70-
70+
7171
It loads the data, splits it into training and testing sets,
7272
trains the regressor on the training data, and evaluates its performance
7373
on the test data.
@@ -95,5 +95,6 @@ def main() -> None:
9595

9696
if __name__ == "__main__":
9797
import doctest
98+
9899
doctest.testmod(verbose=True)
99100
main()

0 commit comments

Comments
 (0)