我是 tf. 我已經(jīng)使用 tensorflow 訓(xùn)練了一個編碼器 - 解碼器。該程序?qū)⒁粋€單詞作為輸入并打印出它的音素。例如:Hello World -> ['h', 'E', 'l', '"', '@U', ' ', 'w', '"', '3`', 'r', ' 5', 'd']我想獲得所選每個音素的預(yù)測概率。在預(yù)測部分,我使用的代碼如下:def predict(words, sess): if len(words) > hp.batch_size: after = predict(words[hp.batch_size:], sess) words = words[:hp.batch_size] else: after = [] x = np.zeros((len(words), hp.maxlen), np.int32) # 0: <PAD> for i, w in enumerate(words): for j, g in enumerate((w + "E")[:hp.maxlen]): x[i][j] = g2idx.get(g, 2) preds = np.zeros((len(x), hp.maxlen), np.int32) for j in range(hp.maxlen): xpreds = sess.run(graph.preds, {graph.x: x, graph.y: preds}) preds[:, j] = xpreds[:, j]提前謝謝你!我的主要問題是這些概率“隱藏”在哪里以及如何訪問它們。例如,單詞“Hello”中的字母“o”被映射到音素“@U”。我想知道“@U”被選為理想音素的概率是多少。
如何在 Tensorflow 中打印出預(yù)測概率
慕的地6264312
2021-08-24 17:05:20