2017-01-23 43 views
1

我使用Tensorflow來執行在線學習和分類,其中沒有提前指定類的數量,即模型必須支持以在線方式添加和刪除類。此外,我計劃將類名(字符串)的字典保存到輸出層中的相應索引。每次添加或刪除課程時,該詞典都會更新。用tensorflow存儲字符串到int字典

有什麼辦法可以用tensorflow圖存儲這個動態字典嗎?我曾看過lookup_ops in tensorflow,但他們不支持添加或刪除類的操作。

回答

0

如何在字符串類型的1d變量中存儲名稱,並使用int類型的第二個1d變量來存儲相應的索引?這將有O(n)查找時間;如果班級數量雖然很少,但可能就足夠了。