我是TensorFlow的新手,正在编写一些代码来解析YT8M数据集。我处于地图功能的最后阶段,我正在尝试将视频标签从其原始标签(超过4,000个)映射到其所属的第一个垂直位置(仅25个)。
以下是一些代码:
def test(self, labels, features):
#Two dictionaries here to do the mappings.
labels = self.vertical_to_int[self.vertical_dict[labels]]
return labels, features
def redo(self):
#get_mapped() is just the first round of mapping that I did
above.
data = self.get_mapped()
return data.map(self.test)
我觉得我已经尽了一切努力来解决此问题。试验了不同版本的Tensorflow,使用了tf.map_fn()
方法而不是尝试内联地进行,尝试执行.numpy()返回EagerTensor的numpy数组,然后对其进行操作。
我不确定从这里去哪里...
谢谢!
!!!请在下面的评论中查看更新!