Commit b618be99 authored by Bharath Ramsundar's avatar Bharath Ramsundar Committed by GitHub
Browse files

Merge pull request #901 from miaecle/IRV 

fix for IRV transformer
parents 75c4bcab 4ea07bdb

deepchem/trans/transformers.py

+1 −0
Original line number Diff line number Diff line
@@ -677,6 +677,7 @@ class IRVTransformer(): 
    values = []

    top_labels = []

    with g_temp.as_default():

      with tf.device('/cpu:0'):

        labels_tf = tf.constant(y)

        similarity_placeholder = tf.placeholder(

            dtype=tf.float64, shape=(None, reference_len))

CRA Git | Maintained and supported by SUSTech CRA and CCSE