Skip to content

Commit 3d45086

Browse files
change
1 parent b3b77a1 commit 3d45086

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

train.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -241,7 +241,8 @@ def main():
241241
for j in range(0,steps-remaining,step_size):
242242
#Feeding step_size-amount data with 0.5 keeping probabilities on DROPOUT LAYERS
243243
_,c = sess.run([train,final_loss],
244-
feed_dict = {x:X[j:j+step_size], y_true_1:Y1[j:j+step_size],y_true_2:Y2[j:j+step_size],hold_prob1:0.5,hold_prob2:0.5})
244+
feed_dict = {x:X[j:j+step_size], y_true_1:Y1[j:j+step_size],
245+
y_true_2:Y2[j:j+step_size],hold_prob1:0.5,hold_prob2:0.5})
245246

246247
#Writing for loop to calculate test statistics. GTX 1050 isn't able to calculate all cv data.
247248
cv_auc_list = []
@@ -252,6 +253,7 @@ def main():
252253
acc_on_cv,loss_on_cv,preds,coordinates = sess.run([acc,cross_entropy,tf.nn.softmax(y_pred_1),y_pred_2],
253254
feed_dict={x:cv_x[v:v+validating_size], y_true_1:cv_y1[v:v+validating_size], y_true_2:cv_y2[v:v+validating_size],
254255
hold_prob1:1.0,hold_prob2:1.0})
256+
255257
auc_on_cv = roc_auc_score(cv_y1[v:v+validating_size],preds)
256258
regression_loss = np.mean(pow(cv_y2[v:v+validating_size] - coordinates , 2 ) )
257259
cv_acc_list.append(acc_on_cv)

0 commit comments

Comments
 (0)