我使用的是一个标准的keras模型,我正在进行批处理培训(使用train_on_batch
函数)。现在,我想获取批处理中每个元素的梯度,并对其进行缩放(将每个样本梯度乘以我所拥有的样本特定值),并且在每个梯度被缩放之后,它可以被求和并用于更新现有的权重。不管怎么说,有这个给定的角函数吗?如果没有,有没有办法让我用tensorflow来操纵这个?(给出了模型,其余的都是用keras编写的)
这个函数如下所示(循环是为了说明它发生在批处理中的所有样本)
grad = 0, w= #array of size batch_size
for i in batch_size:
grad <- grad + w_i*grad_i
发布于 2018-12-06 16:02:25
sample_weights
方法中使用fit
参数。X_train, y_train
,而且返回X_train, y_train, sample_weights
。在这两种情况下,sample_weights
应该是一个一维向量,与数据的样本数相同。
https://stackoverflow.com/questions/53661344
复制