Tensorflow:我可以定义加权均方误差损失函数吗?

时间:2018-01-19 14:26:10

标签: python tensorflow keras

我有一个具有以下形状的数据集:

x1   x2   x3   w   y
1    2    5    3   4
4    6    6    1   2
5    2    7    3   1
1    6    8    1   8
...

我想从yx1x2预测x3,使用以下损失函数(Pyhton中的示例):

loss = numpy.dot(w, (y-y_pred)**2)

是否可以在Tensorflow(或Keras)中训练具有此损失函数的模型?

模型必须仅使用x1x2x3来预测y,而w在推理期间是未知的。

0 个答案:

没有答案