为什么scikit-learn的平均精确度得分返回nan?

我的Keras模型设计用于接收两个输入时间序列,将它们连接起来,通过LSTM层处理,并对下一个时间步进行多标签预测。

有50个训练样本,每个样本有24个时间步,每个样本有5625个标签。

有12个验证样本,每个样本有24个时间步,每个样本有5625个标签。

当我尝试验证模型时,average_precision_score返回了’nan’。这是为什么呢?

我在下面准备了一个最简示例(MWE),来说明这个问题:

这段代码无需翻译

运行上述代码应该会在不到30秒的时间内给出类似这样的结果。请注意平均精确度返回了nan:

这段代码无需翻译

即使使用一个更简单的模型,仅有一个输入,也会出现同样的错误:

这段代码无需翻译

回答:

问题出在错误的(颠倒的)维度上。将矩阵展平解决了这个问题:

y_val_true, val_pred = y_val_true.reshape((-1)), val_pred.reshape((-1))val_ap = average_precision_score(y_val_true, val_pred)

Related Posts

L1-L2正则化的不同系数

我想对网络的权重同时应用L1和L2正则化。然而,我找不…

使用scikit-learn的无监督方法将列表分类成不同组别,有没有办法?

我有一系列实例,每个实例都有一份列表,代表它所遵循的不…

f1_score metric in lightgbm

我想使用自定义指标f1_score来训练一个lgb模型…

通过相关系数矩阵进行特征选择

我在测试不同的算法时,如逻辑回归、高斯朴素贝叶斯、随机…

可以将机器学习库用于流式输入和输出吗?

已关闭。此问题需要更加聚焦。目前不接受回答。 想要改进…

在TensorFlow中,queue.dequeue_up_to()方法的用途是什么?

我对这个方法感到非常困惑,特别是当我发现这个令人费解的…

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注