Update class_DeepHit.py

typo fixed on loss1 computation

Author: chl8856

class_DeepHit.py

@@ -121,7 +121,7 @@ def loss_Log_Likelihood(self):
         tmp1 = I_1 * log(tmp1)
 
         #for censored: log \sum P(T>t|x)
-        tmp2 = tf.reduce_sum(tf.reduce_sum(self.fc_mask1 * self.out, reduction_indices=2), reduction_indices=1, keep_dims=True)
+        tmp2 = tf.reduce_sum(tf.reduce_sum(self.fc_mask2 * self.out, reduction_indices=2), reduction_indices=1, keep_dims=True)
         tmp2 = (1. - I_1) * log(tmp2)
 
         self.LOSS_1 = - tf.reduce_mean(tmp1 + 1.0*tmp2)
@@ -202,4 +202,4 @@ def predict(self, x_test, keep_prob=1.0):
     # def predict(self, x_test, MASK, keep_prob=1.0):
     #     (m1_test, m2_test) = MASK
     #     return self.sess.run(self.out, 
-    #                          feed_dict={self.x: x_test, self.rnn_mask1:m1_test, self.rnn_mask2:m2_test, self.keep_prob: keep_prob})
+    #                          feed_dict={self.x: x_test, self.rnn_mask1:m1_test, self.rnn_mask2:m2_test, self.keep_prob: keep_prob})