Skip to content
Snippets Groups Projects
Commit f4a84d2d authored by Tim O'Donnell's avatar Tim O'Donnell
Browse files

tweak

parent 5cb3b145
No related branches found
No related tags found
No related merge requests found
......@@ -155,7 +155,6 @@ class Class1LigandomePredictor(object):
Loss function for ligandome prediction.
"""
import tensorflow as tf
import keras.backend as K
y_pred = tf.squeeze(y_pred, axis=-1)
y_true = tf.reshape(tf.cast(y_true, tf.bool), (-1,))
......
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