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

snapshot...

parent 63a9e447
Branches
No related tags found
No related merge requests found
...@@ -1122,15 +1122,8 @@ def run_evaluate_static(data_dct, num_tiles, ckpt_dir_s_path, day_night='DAY', l ...@@ -1122,15 +1122,8 @@ def run_evaluate_static(data_dct, num_tiles, ckpt_dir_s_path, day_night='DAY', l
nn.setup_eval_pipeline(data_dct, num_tiles) nn.setup_eval_pipeline(data_dct, num_tiles)
nn.do_evaluate(prob_thresh=prob_thresh) nn.do_evaluate(prob_thresh=prob_thresh)
probs = nn.test_probs probs_dct[flvl] = nn.test_probs.flatten()
preds = nn.test_preds preds_dct[flvl] = nn.test_preds.flatten()
# if NumClasses == 2:
# preds = np.where(probs > prob_thresh, 1, 0)
# else:
# preds = np.argmax(probs, axis=1)
probs_dct[flvl] = probs
preds_dct[flvl] = preds
return preds_dct, probs_dct return preds_dct, probs_dct
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment