Skip to content

Commit 5ad0cda

Browse files
Change
1 parent 7567e39 commit 5ad0cda

File tree

2 files changed

+8
-16
lines changed

2 files changed

+8
-16
lines changed

multiclass_classification_cpu.py

Lines changed: 3 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -241,12 +241,9 @@
241241
cv_acc_list = []
242242
cv_loss_list = []
243243
for v in range(0,len(cv_x)-int(len(cv_x) % validating_size),validating_size):
244-
acc_on_cv,loss_on_cv,preds = sess.run([acc,cross_entropy,tf.nn.softmax(y_pred)],\
245-
feed_dict=\
246-
{x:cv_x[v:v+validating_size] ,\
247-
y_true:cv_y[v:v+validating_size] ,\
248-
hold_prob1:1.0,\
249-
hold_prob2:1.0})
244+
acc_on_cv,loss_on_cv,preds = sess.run([acc,cross_entropy,tf.nn.softmax(y_pred)],
245+
feed_dict={x:cv_x[v:v+validating_size] ,y_true:cv_y[v:v+validating_size] ,hold_prob1:1.0,hold_prob2:1.0})
246+
250247
auc_on_cv = roc_auc_score(cv_y[v:v+validating_size],preds)
251248
cv_acc_list.append(acc_on_cv)
252249
cv_auc_list.append(auc_on_cv)

multiclass_classification_gpu.py

Lines changed: 5 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -243,11 +243,9 @@
243243
cv_acc_list = []
244244
cv_loss_list = []
245245
for v in range(0,len(cv_x)-int(len(cv_x) % validating_size),validating_size):
246-
acc_on_cv,loss_on_cv,preds = sess.run([acc,cross_entropy,tf.nn.softmax(y_pred)],feed_dict=
247-
{x:cv_x[v:v+validating_size] ,
248-
y_true:cv_y[v:v+validating_size] ,
249-
hold_prob1:1.0,
250-
hold_prob2:1.0})
246+
acc_on_cv,loss_on_cv,preds = sess.run([acc,cross_entropy,tf.nn.softmax(y_pred)],
247+
feed_dict={x:cv_x[v:v+validating_size] ,y_true:cv_y[v:v+validating_size] ,hold_prob1:1.0,hold_prob2:1.0})
248+
251249
auc_on_cv = roc_auc_score(cv_y[v:v+validating_size],preds)
252250
cv_acc_list.append(acc_on_cv)
253251
cv_auc_list.append(auc_on_cv)
@@ -264,11 +262,8 @@
264262
test_acc_list = []
265263
test_loss_list = []
266264
for v in range(0,len(test_x)-int(len(test_x) % validating_size),validating_size):
267-
acc_on_test,loss_on_test,preds = sess.run([acc,cross_entropy,tf.nn.softmax(y_pred)],feed_dict=\
268-
{x:test_x[v:v+validating_size] ,\
269-
y_true:test_y[v:v+validating_size] ,\
270-
hold_prob1:1.0,\
271-
hold_prob2:1.0})
265+
acc_on_test,loss_on_test,preds = sess.run([acc,cross_entropy,tf.nn.softmax(y_pred)],
266+
feed_dict={x:test_x[v:v+validating_size] ,y_true:test_y[v:v+validating_size] ,hold_prob1:1.0,hold_prob2:1.0})
272267

273268
auc_on_test = roc_auc_score(test_y[v:v+validating_size],preds)
274269
test_acc_list.append(acc_on_test)

0 commit comments

Comments
 (0)