提交 6cbf95dd 编写于 作者: S ShawnXuan

ndarray -> numpy

上级 7595690b
......@@ -182,9 +182,9 @@ def main():
eval_loss = 0.0
for j in range(FLAGS.eval_batchs):
loss, pred, ref = eval_job().get()
label_ = ref.ndarray().astype(np.float32)
label_ = ref.numpy().astype(np.float32)
labels = np.concatenate((labels, label_), axis=0)
preds = np.concatenate((preds, pred.ndarray()), axis=0)
preds = np.concatenate((preds, pred.numpy()), axis=0)
eval_loss += loss.mean()
auc = roc_auc_score(labels[1:], preds[1:])
print(i+1, "eval_loss", eval_loss/FLAGS.eval_batchs, "eval_auc", auc)
......
......@@ -205,9 +205,9 @@ def main():
eval_loss = 0.0
for i in range(eval_epoch_size):
loss, pred, ref = eval_job().get()
label_ = ref.ndarray().astype(np.float32)
label_ = ref.numpy().astype(np.float32)
labels = np.concatenate((labels, label_), axis=0)
preds = np.concatenate((preds, pred.ndarray()), axis=0)
preds = np.concatenate((preds, pred.numpy()), axis=0)
eval_loss += loss.mean()
auc = roc_auc_score(labels[1:], preds[1:])
print(epoch, "eval_loss", eval_loss/eval_epoch_size, "eval_auc", auc)
......@@ -217,9 +217,9 @@ def main():
eval_loss = 0.0
for i in range(test_epoch_size):
loss, pred, ref = test_job().get()
label_ = ref.ndarray().astype(np.float32)
label_ = ref.numpy().astype(np.float32)
labels = np.concatenate((labels, label_), axis=0)
preds = np.concatenate((preds, pred.ndarray()), axis=0)
preds = np.concatenate((preds, pred.numpy()), axis=0)
eval_loss += loss.mean()
auc = roc_auc_score(labels[1:], preds[1:])
print("test_loss", eval_loss/test_epoch_size, "eval_auc", auc)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册