Toggle navigation
Toggle navigation
This project
Loading...
Sign in
Victoria BOURGEAIS
/
DeepGONet
Go to a project
Toggle navigation
Toggle navigation pinning
Projects
Groups
Snippets
Help
Project
Activity
Repository
Pipelines
Graphs
Issues
0
Merge Requests
0
Wiki
Network
Create a new issue
Builds
Commits
Authored by
Victoria BOURGEAIS
2020-11-27 14:10:20 +0100
Browse Files
Options
Browse Files
Download
Email Patches
Plain Diff
Commit
47932faa9f7985943772b562b84e43f01be6614b
47932faa
1 parent
369ab85a
remove an error in the multiclassifier
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
2 additions
and
1 deletions
.gitignore
DeepGONet.py
.gitignore
View file @
47932fa
...
...
@@ -2,3 +2,4 @@ X_test.npz
X_train.npz
filesforNNarch.zip
__pycache__/
.ipynb_checkpoints/
...
...
DeepGONet.py
View file @
47932fa
...
...
@@ -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
=
t
f
.
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
))
...
...
Please
register
or
login
to post a comment