Skip to content

Commit 5f9cae8

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 80d86c1 commit 5f9cae8

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

machine_learning/xgboostclassifier.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -8,9 +8,9 @@
88

99
def data_handling(data: dict) -> tuple:
1010
# Split dataset into train and test data
11-
features = data["data"] # data is features
11+
features = data["data"] # data is features
1212
target = data["target"]
13-
x=(train_test_split(features, target, test_size=0.25))
13+
x = train_test_split(features, target, test_size=0.25)
1414
return x
1515

1616

@@ -33,7 +33,7 @@ def main() -> None:
3333

3434
names = iris["target_names"]
3535

36-
x_train, x_test, y_train, y_test = data_handling(iris)
36+
x_train, x_test, y_train, y_test = data_handling(iris)
3737

3838
# XGBoost Classifier
3939
xgb = xgboost(x_train, y_train)

0 commit comments

Comments
 (0)