change str in config (act, agg) from str to callable

This commit is contained in:
wls2002
2023-08-05 03:03:02 +08:00
parent 0e44b13291
commit af54db3b12
10 changed files with 55 additions and 101 deletions

View File

@@ -1,5 +1,5 @@
from dataclasses import dataclass
from utils import Act, Agg
@dataclass(frozen=True)
class BasicConfig:
@@ -68,8 +68,8 @@ class NeatConfig:
class HyperNeatConfig:
below_threshold: float = 0.2
max_weight: float = 3
activation: str = "sigmoid"
aggregation: str = "sum"
activation: callable = Act.sigmoid
aggregation: callable = Agg.sum
activate_times: int = 5
inputs: int = 2
outputs: int = 1