Skip to content
Snippets Groups Projects
Commit 11b1d4c7 authored by Noordijk, Ben's avatar Noordijk, Ben
Browse files

Set logit cutoff to 5 instead of 0

parent b50c353b
No related branches found
No related tags found
1 merge request!5compare_accuracy.py now saves confusion matrices and can be called on the...
......@@ -24,7 +24,8 @@ class InferenceModel(object):
self.batch_size = batch_size
def predict(self, read, kmer):
return np.any(self._model_dict[kmer](read) > 0.0) # todo make threshold feature?
model_output = self._model_dict[kmer](read)
return np.any(model_output > 5.0) # todo make threshold feature?
# # For fixed batch sizes
# read, last_idx = read
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment