Victoria BOURGEAIS / DeepGONet

Go to a project
Toggle navigation Toggle navigation pinning
Victoria BOURGEAIS
47932faa 1 parent 369ab85a

remove an error in the multiclassifier

Inline Side-by-side
Showing 2 changed files with 2 additions and 1 deletions
...@@ -2,3 +2,4 @@ X_test.npz ...@@ -2,3 +2,4 @@ X_test.npz
2 X_train.npz 2 X_train.npz
3 filesforNNarch.zip 3 filesforNNarch.zip
4 __pycache__/ 4 __pycache__/
5 +.ipynb_checkpoints/
......
...@@ -142,7 +142,7 @@ def train(save_dir): ...@@ -142,7 +142,7 @@ def train(save_dir):
142 # ---- CE loss ---- 142 # ---- CE loss ----
143 # Compute the average of the loss across all the dimensions 143 # Compute the average of the loss across all the dimensions
144 if FLAGS.n_classes>=2: 144 if FLAGS.n_classes>=2:
145 - ce_loss = f.reduce_mean(tf.nn.softmax_cross_entropy_with_logits_v2(logits=pred, labels=Y)) 145 + ce_loss = tf.reduce_mean(tf.nn.softmax_cross_entropy_with_logits_v2(logits=pred, labels=Y))
146 else: 146 else:
147 ce_loss = tf.reduce_mean(tf.nn.sigmoid_cross_entropy_with_logits(logits=pred, labels=Y)) 147 ce_loss = tf.reduce_mean(tf.nn.sigmoid_cross_entropy_with_logits(logits=pred, labels=Y))
148 148
......