亚洲在线久爱草,狠狠天天香蕉网,天天搞日日干久草,伊人亚洲日本欧美

為了賬號安全,請及時綁定郵箱和手機立即綁定
已解決430363個問題,去搜搜看,總會有你想問的

如何將 tf 模型預測與層相關聯?

如何將 tf 模型預測與層相關聯?

互換的青春 2023-06-13 17:11:10
我們正在研究我們的Tensorflow人工智能Sequential model,它具有輸入數據數組,并提供預測“真”或“假”的概率。我們想知道哪個預測對應于“真”,哪個對應于“假”我們的模型:model = tf.keras.Sequential([                          layers..        tf.keras.layers.GlobalAveragePooling1D(),        tf.keras.layers.Dense(2, activation='softmax')    ])模型編譯:model.compile(loss="sparse_categorical_crossentropy",                  optimizer=tf.keras.optimizers.Adam(),                  metrics=['accuracy']                  )然后模型適合tests和results,其中results是“真”或“假”。history = self.model.fit(        np.array(self.data["tests"], dtype=float),        np.array(self.data["results"], dtype=float),        validation_split=0.1,        epochs=self.epochs,        batch_size=self.batch_size,        steps_per_epoch=self.steps_per_epoch,        verbose=0,        shuffle=True,        callbacks=[PlotLossesKerasTF()],    )當我們對我們使用的新數據進行預測時model.predict(newData),它會為我們提供如下概率:[[0.5787903  0.42120975]]那么這些數字中的哪些對應于哪個標簽?
查看完整描述

1 回答

?
holdtom

TA貢獻1805條經驗 獲得超10個贊

它與您的訓練標簽相同。

假設您有一張輸入圖像,您可以對它是貓、狗還是鳥進行分類。然后,您的訓練數據是一組圖像和相應的標簽,格式[cat dog bird]為例如 - 狗標簽的圖像[0 1 0]。所以以model.predict(image)相同的格式輸出概率數組[cat dog bird],output[0]貓分類的概率等等。

此外,如果您的輸出是真/假,請考慮使用一個具有 sigmoid 激活和二元交叉熵的神經元輸出作為您的損失函數。


查看完整回答
反對 回復 2023-06-13
  • 1 回答
  • 0 關注
  • 111 瀏覽
慕課專欄
更多

添加回答

舉報

0/150
提交
取消
微信客服

購課補貼
聯系客服咨詢優惠詳情

幫助反饋 APP下載

慕課網APP
您的移動學習伙伴

公眾號

掃描二維碼
關注慕課網微信公眾號