我們正在研究我們的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 激活和二元交叉熵的神經元輸出作為您的損失函數。
添加回答
舉報
0/150
提交
取消