神经网络-将模型输出转换为预测的目标类

import numpy as np

def predict_class(model_output, threshold=0.5):
    """
    将模型输出转换为预测的目标类
    :param model_output: 模型输出，可以是概率分布或概率值
    :param threshold: 阈值，用于二分类问题
    :return: 预测的目标类
    """
    if len(model_output) == 1:  # 二分类问题
        return 1 if model_output[0] > threshold else 0
    else:  # 多分类问题
        return np.argmax(model_output)

# 示例
model_output_binary = np.array([0.7])  # 二分类问题示例
model_output_multiclass = np.array([0.2, 0.3, 0.5])  # 多分类问题示例

predicted_class_binary = predict_class(model_output_binary, threshold=0.6)
predicted_class_multiclass = predict_class(model_output_multiclass)

print("二分类预测结果:", predicted_class_binary)
print("多分类预测结果:", predicted_class_multiclass)