AI开发平台MODELARTS-多层感知机分类(pytorch):参数说明

时间：2024-05-28 16:22:36

AI开发平台MODELARTS 分类

参数	是否必选	参数说明	默认值
feature_index_list	是	feature列的index，不同index之间以','分割。举例： 1,2,3,4	""
label_index	是	label列的index。举例：5	""
hidden_layer_list	是	隐藏层神经元的个数,不同数值之间以','分割，每个数值代表每一层神经元的个数。 int类型，范围[]	"10,5"
hidden_layer_activation	是	隐藏层激活函数，可选；范围['Sigmoid','ReLU']	Sigmoid
epochs	是	训练迭代次数,int类型，范围[]	4
batch_size	是	batch_size大小，int类型，范围[]	24
learning_rate	是	学习率，double类型，范围[0,10]	0.01