https://blog.youkuaiyun.com/Fwuyi/article/details/123213437
深度学习模型在喂数据进行训练前常会用到compile函数进行训练时所使用优化器、损失函数等的配置 。
1.model.compile() 作用
设置优化器、损失函数和准确率评测标准。
2.model.compile() 语法
形式1:
model.compile(optimizer = tf.keras.optimizers.优化器(参数),
loss = tf.keras.losses.损失函数(参数),
metrics = [“sparse_accuracy”])
形式2(常用):
model.compile(optimizer = “sgd”,
loss = “mse”,
metrics = [“sparse_accuracy”])
3.参数设置
3.1 optimizer:预定义优化器名或优化器对象。
(1).“sgd” 或者 tf.optimizers.SGD(lr = 学习率, decay = 学习率衰减率,momentum = 动量参数)
(2).“adagrad” 或者 tf.keras.optimizers.Adagrad(lr = 学习率, decay = 学习率衰减率)
(3).“adadelta” 或者 tf.keras.optimizers.Adadelta(lr = 学习率,decay = 学习率衰减率)
(4).“adam” 或者 tf.keras.optimizers.Adam(lr = 学习率, decay = 学习率衰减率)
3.2 loss:预定义损失函数名或一个目标函数。
(1).“mse” 或者 “mean squared error” 或者 tf.keras.losses.MeanSquaredError()
(2).“sparse_categorical_crossentropy” 或 tf.keras.losses.SparseCatagoricalCrossentropy(from_logits = False)
3.3 Metrics:评估模型在训练和测试时的性能的指标。
(1).“accuracy” :
(2).“sparse_accuracy":
(3).“sparse_categorical_accuracy” :

该博客详细介绍了在深度学习中,如何使用`model.compile()`来配置优化器、损失函数和评估指标。`model.compile()`的主要作用是设定模型训练过程中的关键参数,如使用SGD、Adagrad、Adadelta或Adam等优化器,以及mse、sparse_categorical_crossentropy等损失函数。此外,它还用于设置如accuracy或sparse_accuracy等性能指标,以便在训练和测试阶段评估模型表现。
2万+

被折叠的 条评论
为什么被折叠?



