Skip to content
Snippets Groups Projects
Commit e19a6cc5 authored by tomrink's avatar tomrink
Browse files

snapshot...

parent 4fb4d92d
Branches
No related tags found
No related merge requests found
......@@ -761,7 +761,7 @@ def run_restore_static(directory, ckpt_dir, out_file=None):
labels, preds = nn.run_restore(directory, ckpt_dir)
if out_file is not None:
np.save(out_file,
[np.squeeze(labels), preds.argmax(axis=3), preds[:, :, :, 0], preds[:, :, :, 1], preds[:, :, :, 2]])
[np.squeeze(labels), preds.argmax(axis=3)])
def run_evaluate_static(in_file, out_file, ckpt_dir):
......@@ -816,9 +816,6 @@ def analyze_3cat(file):
tup = np.load(file, allow_pickle=True)
lbls = tup[0]
pred = tup[1]
prob_0 = tup[2]
prob_1 = tup[3]
prob_2 = tup[4]
lbls = lbls.flatten()
pred = pred.flatten()
......@@ -918,9 +915,6 @@ def analyze_5cat(file='/Users/tomrink/cld_opd_frac.npy'):
tup = np.load(file, allow_pickle=True)
lbls = tup[0]
pred = tup[1]
# prob_0 = tup[2]
# prob_1 = tup[3]
# prob_2 = tup[4]
lbls = lbls.flatten()
pred = pred.flatten()
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment