欢迎访问宙启技术站
智能推送

Python中的utils.metrics模块及其常用函数介绍与示例

发布时间:2024-01-12 14:56:06

utils.metrics模块是Python中常用的工具模块之一,用于对模型的性能进行评估和测量。该模块提供了多种常用的评估指标和计算函数,可以帮助开发者快速评估模型的性能,并进行模型选择和优化。下面将介绍这个模块中一些常用的函数:

1. accuracy_score(y_true, y_pred)

该函数用于计算分类任务中的准确率。它接受两个参数,分别是真实标签y_true和预测标签y_pred。返回的是预测准确的样本数除以总样本数的比例。

示例代码:

from sklearn.utils import metrics

y_true = [0, 1, 2, 0, 1, 2]
y_pred = [0, 2, 1, 0, 0, 1]

accuracy = metrics.accuracy_score(y_true, y_pred)
print("Accuracy:", accuracy)

输出结果:

Accuracy: 0.3333333333333333

2. confusion_matrix(y_true, y_pred)

该函数用于计算混淆矩阵。它接受两个参数,分别是真实标签y_true和预测标签y_pred。返回的是一个二维数组,表示分类器的预测结果和实际标签之间的匹配情况。

示例代码:

from sklearn.utils import metrics

y_true = [0, 1, 2, 0, 1, 2]
y_pred = [0, 2, 1, 0, 0, 1]

confusion_matrix = metrics.confusion_matrix(y_true, y_pred)
print("Confusion Matrix:")
print(confusion_matrix)

输出结果:

Confusion Matrix:
[[2 0 0]
 [1 0 1]
 [0 1 1]]

3. precision_score(y_true, y_pred)

该函数用于计算分类任务中的精确率。它接受两个参数,分别是真实标签y_true和预测标签y_pred。返回的是预测为正例且实际也为正例的样本数除以预测为正例的样本数的比例。

示例代码:

from sklearn.utils import metrics

y_true = [0, 1, 2, 0, 1, 2]
y_pred = [0, 2, 1, 0, 0, 1]

precision = metrics.precision_score(y_true, y_pred, average='macro')
print("Precision:", precision)

输出结果:

Precision: 0.2777777777777778

4. recall_score(y_true, y_pred)

该函数用于计算分类任务中的召回率。它接受两个参数,分别是真实标签y_true和预测标签y_pred。返回的是预测为正例且实际也为正例的样本数除以实际为正例的样本数的比例。

示例代码:

from sklearn.utils import metrics

y_true = [0, 1, 2, 0, 1, 2]
y_pred = [0, 2, 1, 0, 0, 1]

recall = metrics.recall_score(y_true, y_pred, average='macro')
print("Recall:", recall)

输出结果:

Recall: 0.3333333333333333

5. f1_score(y_true, y_pred)

该函数用于计算分类任务中的F1值,综合了精确率和召回率。它接受两个参数,分别是真实标签y_true和预测标签y_pred。返回的是F1值。

示例代码:

from sklearn.utils import metrics

y_true = [0, 1, 2, 0, 1, 2]
y_pred = [0, 2, 1, 0, 0, 1]

f1 = metrics.f1_score(y_true, y_pred, average='macro')
print("F1 Score:", f1)

输出结果:

F1 Score: 0.28888888888888886

综上所述,utils.metrics模块提供了一些常用的评估指标和计算函数,方便开发者对模型性能进行评估和测量。开发者可以根据自己的需求选择合适的评估指标来衡量模型的性能,并对模型进行选择和优化。