y_preds.append(preds.numpy())和 y_preds.append(tf.sigmoid(preds).numpy().argmax(axis=1).tolist())的区别
时间: 2024-06-06 18:09:14 浏览: 169
inception_v3.rar
y_preds.append(preds.numpy())将一个TensorFlow张量对象的数值转化成NumPy 数组格式,而y_preds.append(tf.sigmoid(preds).numpy().argmax(axis=1).tolist())则使用Sigmoid函数对张量中的数值进行操作,将其转化成概率值并取得概率最大的标签,然后将标签转换为列表格式。两者的输出结果不同,前者输出张量对象的数值,后者输出一个标签列表。
阅读全文