TensorFlow TypeError:预期的二进制或Unicode字符串,得到<tf.Tensor'Const_1006:0'shape =(15,)dtype = int64>

时间:2019-12-20 00:31:15

标签: python tensorflow hashmap hashtable

我正在尝试在TensorFlow中构造一个哈希表。我需要在数据框中查找特定的行,并将该行的功能输入到我的网络中。我不明白为什么会收到此错误。 docs指定输入为张量流常量,而不是二进制或字符串。

https://github.com/tensorflow/tensorflow/blob/r2.0/tensorflow/python/ops/lookup_ops.py#L262-L291

tab = pd.read_csv('./cohortCovar.tsv', sep='\t', index_col=0)
pid = list(tab.index)
tid = [tf.convert_to_tensor(x) for x in pid]
tab.reset_index(drop=True)
keys = tid
vals = []
for i in range(len(pid)):
  trow = tf.convert_to_tensor(tab.iloc[i])
  vals.append(trow)
values = np.array(vals)

tb = tf.lookup.StaticHashTable(tf.lookup.KeyValueTensorInitializer(keys, values), -1)

0 个答案:

没有答案