DIEN模型的loss为负数
nixekgnaw opened this issue · 0 comments
DIEN模型bug?我使用sampledata进行测试,loss一直都是负数,感觉不太对,不确定这是什么带来的
tensorflow运行结果
Epoch 1/5
/usr/local/lib/python3.7/dist-packages/tensorflow/python/keras/engine/functional.py:591: UserWarning: Input dict contained keys ['rating', 'timestamp', 'userAvgReleaseYear', 'userReleaseYearStddev'] which did not match any model input. They will be ignored by the model.
[n for n in tensors.keys() if n not in ref_input_names])
7403/7403 [==============================] - 175s 23ms/step - loss: 3.4184 - auc_6: 0.5772 - auc_value: 0.5565
Epoch 2/5
7403/7403 [==============================] - 162s 22ms/step - loss: -3.2109 - auc_6: 0.6725 - auc_value: 0.6455
Epoch 3/5
7403/7403 [==============================] - 162s 22ms/step - loss: -3.4127 - auc_6: 0.7511 - auc_value: 0.7350
Epoch 4/5
7403/7403 [==============================] - 162s 22ms/step - loss: -3.4589 - auc_6: 0.7959 - auc_value: 0.7831
Epoch 5/5
7403/7403 [==============================] - 162s 22ms/step - loss: -3.4888 - auc_6: 0.8209 - auc_value: 0.8140
1870/1870 [==============================] - 23s 11ms/step - loss: -3.3402 - auc_6: 0.7502 - auc_value: 0.7512
Test Loss -3.3401615619659424, Test Accuracy 0.7501789331436157, Test ROC AUC 0.7511806488037109,
以及DIEN.py的304行报错,
test_loss, test_roc_auc = model.evaluate(test_dataset)
ValueError: too many values to unpack (expected 2)
感觉应该是
test_loss, test_roc_auc ,test_accuracy = model.evaluate(test_dataset)