This commit is contained in:
parent
70431cad4f
commit
324a2245a7
|
|
@ -22,7 +22,7 @@ class NeuralNetwork:
|
||||||
structure: List[int],
|
structure: List[int],
|
||||||
hidden_activate: Literal["relu"] = "relu",
|
hidden_activate: Literal["relu"] = "relu",
|
||||||
output_activate: Literal["linear", "softmax"] = "linear",
|
output_activate: Literal["linear", "softmax"] = "linear",
|
||||||
seed: int = 52,
|
seed: int = 62,
|
||||||
epsilon: float = 1e-9,
|
epsilon: float = 1e-9,
|
||||||
):
|
):
|
||||||
"""
|
"""
|
||||||
|
|
@ -30,7 +30,7 @@ class NeuralNetwork:
|
||||||
:param structure: 神经网络结构,例如[2, 10, 1]表示2层神经网络,具体为输入层2个神经元、隐含层10个神经元、输出层1个神经元
|
:param structure: 神经网络结构,例如[2, 10, 1]表示2层神经网络,具体为输入层2个神经元、隐含层10个神经元、输出层1个神经元
|
||||||
:param hidden_activate: 隐含层的激活函数,默认为relu
|
:param hidden_activate: 隐含层的激活函数,默认为relu
|
||||||
:param output_activate: 输出层的激活函数,默认为linear
|
:param output_activate: 输出层的激活函数,默认为linear
|
||||||
:param seed: 随机种子,默认为52
|
:param seed: 随机种子
|
||||||
:param epsilon: 极小值,默认为1e-9
|
:param epsilon: 极小值,默认为1e-9
|
||||||
"""
|
"""
|
||||||
print("正在初始化神经网络...", end="")
|
print("正在初始化神经网络...", end="")
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue