在Python中统计正确率通常涉及以下步骤:
1. 导入必要的库和模块,如`numpy`和`sklearn.metrics`中的`accuracy_score`函数。
2. 加载测试数据集和预测结果。
3. 比较预测结果和实际标签,计算准确率。
4. 打印准确率结果。
导入必要的库和模块
import numpy as np
from sklearn.metrics import accuracy_score
加载测试数据集和预测结果
test_labels = np.load("test_labels.npy") 实际标签
predicted_labels = np.load("predicted_labels.npy") 预测结果
计算准确率
accuracy = accuracy_score(test_labels, predicted_labels)
打印准确率结果
print("准确率:", accuracy)
如果你是在神经网络模型训练过程中计算准确率,可以使用`torch`库中的`accuracy_score`函数,如下所示:
导入必要的库和模块
import torch
from torch import nn
from torch.utils.data import DataLoader
假设model是你的模型,train_loader是训练数据加载器
optimizer = torch.optim.Adam(model.parameters())
loss_func = nn.CrossEntropyLoss()
EPOCHS = 5
for epoch in range(EPOCHS):
correct = 0
for batch_idx, (X_batch, y_batch) in enumerate(train_loader):
optimizer.zero_grad()
output = model(X_batch.float())
loss = loss_func(output, y_batch)
loss.backward()
optimizer.step()
计算准确率
predicted = torch.max(output.data, 1)
correct += (predicted == y_batch).sum().item()
计算当前epoch的准确率
train_accuracy = correct / len(train_loader.dataset)
print(f"Epoch {epoch+1} train accuracy: {train_accuracy}")
请注意,这些代码示例可能需要根据你的具体情况进行调整。