Update app.py
Browse files
app.py
CHANGED
@@ -129,7 +129,7 @@ class Network(object):
|
|
129 |
predicted_output = predicted_output.reshape((self.G, self.G, self.G), order='F')
|
130 |
|
131 |
# return idx, other_data_input, true_output, predicted_output
|
132 |
-
return
|
133 |
|
134 |
def get_geometry(self, idx=None):
|
135 |
|
|
|
129 |
predicted_output = predicted_output.reshape((self.G, self.G, self.G), order='F')
|
130 |
|
131 |
# return idx, other_data_input, true_output, predicted_output
|
132 |
+
return predicted_output
|
133 |
|
134 |
def get_geometry(self, idx=None):
|
135 |
|