Victoria BOURGEAIS

remove an error in the multiclassifier

......@@ -2,3 +2,4 @@ X_test.npz
X_train.npz
filesforNNarch.zip
__pycache__/
.ipynb_checkpoints/
......
......@@ -142,7 +142,7 @@ def train(save_dir):
# ---- CE loss ----
# Compute the average of the loss across all the dimensions
if FLAGS.n_classes>=2:
ce_loss = f.reduce_mean(tf.nn.softmax_cross_entropy_with_logits_v2(logits=pred, labels=Y))
ce_loss = tf.reduce_mean(tf.nn.softmax_cross_entropy_with_logits_v2(logits=pred, labels=Y))
else:
ce_loss = tf.reduce_mean(tf.nn.sigmoid_cross_entropy_with_logits(logits=pred, labels=Y))
......