Skip to content

Commit 83ad998

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

1 file changed

Lines changed: 10 additions & 2 deletions

File tree

machine_learning/mlp_activation_comparison.py

Lines changed: 10 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,13 +5,21 @@
55
from sklearn.metrics import accuracy_score
66

77
X, y = make_moons(n_samples=500, noise=0.2, random_state=42)
8-
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.3, random_state=42)
8+
X_train, X_test, y_train, y_test = train_test_split(
9+
X, y, test_size=0.3, random_state=42
10+
)
911

1012
activations = ["logistic", "tanh", "relu"]
1113
results = {}
1214

1315
for act in activations:
14-
clf = MLPClassifier(hidden_layer_sizes=(10, 5), activation=act, solver="adam", max_iter=1000, random_state=42)
16+
clf = MLPClassifier(
17+
hidden_layer_sizes=(10, 5),
18+
activation=act,
19+
solver="adam",
20+
max_iter=1000,
21+
random_state=42,
22+
)
1523
clf.fit(X_train, y_train)
1624
y_pred = clf.predict(X_test)
1725
acc = accuracy_score(y_test, y_pred)

0 commit comments

Comments
 (0)