Skip to content

Commit ea49a93

Browse files
committed
Fix errors
1 parent 2703cdd commit ea49a93

File tree

4 files changed

+5
-6
lines changed

4 files changed

+5
-6
lines changed

abfo_mlnn_script.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -32,14 +32,14 @@ def train_model(item):
3232
"path_save_result": pathsave + requirement_variables[4],
3333
"test_type": test_type,
3434
"draw": True,
35-
"print_train": 0 # 0: nothing, 1 : full detail, 2: short version
35+
"print_train": 1 # 0: nothing, 1 : full detail, 2: short version
3636
}
3737
root_hybrid_paras = {
3838
"hidden_size": item["hidden_size"], "activations": item["activations"], "epoch": item["epoch"],
3939
"train_valid_rate": item["train_valid_rate"], "domain_range": item["domain_range"]
4040
}
4141
abfols_paras = {
42-
"pop_size": item["pop_size"], "Ci": item["Ci"], "Ped": item["Ped"], "Ns": item["Ns"],
42+
"epoch": item["epoch"], "pop_size": item["pop_size"], "Ci": item["Ci"], "Ped": item["Ped"], "Ns": item["Ns"],
4343
"N_minmax": item["N_minmax"]
4444
}
4545
md = ABfoLSMlnn(root_base_paras=root_base_paras, root_hybrid_paras=root_hybrid_paras, abfols_paras=abfols_paras)

cro_mlnn_script.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,7 @@ def train_model(item):
3333
"path_save_result": pathsave + requirement_variables[4],
3434
"test_type": test_type,
3535
"draw": True,
36-
"print_train": 0 # 0: nothing, 1 : full detail, 2: short version
36+
"print_train": 1 # 0: nothing, 1 : full detail, 2: short version
3737
}
3838
root_hybrid_paras = {
3939
"hidden_size": item["hidden_size"], "activations": item["activations"], "epoch": item["epoch"],

ga_mlnn_script.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,7 @@ def train_model(item):
3333
"path_save_result": pathsave + requirement_variables[4],
3434
"test_type": test_type,
3535
"draw": True,
36-
"print_train": 0 # 0: nothing, 1 : full detail, 2: short version
36+
"print_train": 1 # 0: nothing, 1 : full detail, 2: short version
3737
}
3838
root_hybrid_paras = {
3939
"hidden_size": item["hidden_size"], "activations": item["activations"], "epoch": item["epoch"],

model/main/hybrid_mlnn.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,5 @@
1-
from model.optimizer.swarm import BFO, ABC, PSO, WOA
1+
from model.optimizer.swarm import BFO, PSO
22
from model.optimizer.evolutionary import GA, DE, CRO
3-
from model.optimizer.physics import QSO
43
from model.root.hybrid.root_hybrid_mlnn import RootHybridMlnn
54

65

0 commit comments

Comments
 (0)