Skip to content

Commit 2cb6276

Browse files
committed
Update examples for mha-based MLP models
1 parent 632dcc6 commit 2cb6276

File tree

3 files changed

+6
-6
lines changed

3 files changed

+6
-6
lines changed

examples/exam_mha_mlp_binary_classification.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,8 +4,8 @@
44
# Github: https://github.com/thieu1995 %
55
# --------------------------------------------------%
66

7-
from metaperceptron import Data, MhaMlpClassifier
87
from sklearn.datasets import load_breast_cancer
8+
from metaperceptron import Data, MhaMlpClassifier
99

1010

1111
## Load data object
@@ -28,7 +28,7 @@
2828
print(MhaMlpClassifier.SUPPORTED_CLS_OBJECTIVES)
2929

3030
model = MhaMlpClassifier(hidden_layers=(100,), act_names="ELU", dropout_rates=None, act_output=None,
31-
optim="BaseGA", optim_paras=opt_paras, obj_name="F1S", seed=42, verbose=True)
31+
optim="BaseGA", optim_params=opt_paras, obj_name="F1S", seed=42, verbose=True)
3232
## Train the model
3333
model.fit(X=data.X_train, y=data.y_train)
3434

examples/exam_mha_mlp_multi_class_classification.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,8 +4,8 @@
44
# Github: https://github.com/thieu1995 %
55
# --------------------------------------------------%
66

7-
from metaperceptron import Data, MhaMlpClassifier
87
from sklearn.datasets import load_iris
8+
from metaperceptron import Data, MhaMlpClassifier
99

1010

1111
## Load data object
@@ -26,7 +26,7 @@
2626
## Create model
2727
opt_paras = {"name": "WOA", "epoch": 100, "pop_size": 20}
2828
model = MhaMlpClassifier(hidden_layers=(50,), act_names="Tanh", dropout_rates=None, act_output=None,
29-
optim="BaseGA", optim_paras=opt_paras, obj_name="F1S", seed=42, verbose=True)
29+
optim="BaseGA", optim_params=opt_paras, obj_name="F1S", seed=42, verbose=True)
3030
## Train the model
3131
model.fit(X=data.X_train, y=data.y_train, lb=-1., ub=1.0)
3232

examples/exam_mha_mlp_regression.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,8 +5,8 @@
55
# --------------------------------------------------%
66

77
import numpy as np
8-
from metaperceptron import Data, MhaMlpRegressor
98
from sklearn.datasets import load_diabetes
9+
from metaperceptron import Data, MhaMlpRegressor
1010

1111

1212
## Load data object
@@ -27,7 +27,7 @@
2727
## Create model
2828
opt_paras = {"name": "WOA", "epoch": 250, "pop_size": 30}
2929
model = MhaMlpRegressor(hidden_layers=(30, 15,), act_names="ELU", dropout_rates=0.2, act_output=None,
30-
optim="BaseGA", optim_paras=opt_paras, obj_name="MSE", seed=42, verbose=True)
30+
optim="BaseGA", optim_params=opt_paras, obj_name="MSE", seed=42, verbose=True)
3131
## Train the model
3232
model.fit(data.X_train, data.y_train)
3333

0 commit comments

Comments
 (0)