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

ndarray -> numpy

上级 7595690b
...@@ -182,9 +182,9 @@ def main(): ...@@ -182,9 +182,9 @@ def main():
eval_loss = 0.0 eval_loss = 0.0
for j in range(FLAGS.eval_batchs): for j in range(FLAGS.eval_batchs):
loss, pred, ref = eval_job().get() 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) 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() eval_loss += loss.mean()
auc = roc_auc_score(labels[1:], preds[1:]) auc = roc_auc_score(labels[1:], preds[1:])
print(i+1, "eval_loss", eval_loss/FLAGS.eval_batchs, "eval_auc", auc) print(i+1, "eval_loss", eval_loss/FLAGS.eval_batchs, "eval_auc", auc)
......
...@@ -205,9 +205,9 @@ def main(): ...@@ -205,9 +205,9 @@ def main():
eval_loss = 0.0 eval_loss = 0.0
for i in range(eval_epoch_size): for i in range(eval_epoch_size):
loss, pred, ref = eval_job().get() 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) 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() eval_loss += loss.mean()
auc = roc_auc_score(labels[1:], preds[1:]) auc = roc_auc_score(labels[1:], preds[1:])
print(epoch, "eval_loss", eval_loss/eval_epoch_size, "eval_auc", auc) print(epoch, "eval_loss", eval_loss/eval_epoch_size, "eval_auc", auc)
...@@ -217,9 +217,9 @@ def main(): ...@@ -217,9 +217,9 @@ def main():
eval_loss = 0.0 eval_loss = 0.0
for i in range(test_epoch_size): for i in range(test_epoch_size):
loss, pred, ref = test_job().get() 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) 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() eval_loss += loss.mean()
auc = roc_auc_score(labels[1:], preds[1:]) auc = roc_auc_score(labels[1:], preds[1:])
print("test_loss", eval_loss/test_epoch_size, "eval_auc", auc) 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.
先完成此消息的编辑!
想要评论请 注册