用Pytorch搭建一个全连接神经网络并使用DataLoader,实现二分类任务,并测试模型并利用f1评估模型
时间: 2024-03-11 10:49:34 浏览: 120
好的,以下是用PyTorch搭建一个全连接神经网络并使用DataLoader实现二分类任务的示例代码:
```python
import torch
import torch.nn as nn
import torch.optim as optim
from torch.utils.data import DataLoader, Dataset
from sklearn.metrics import f1_score
# 定义一个自定义数据集类
class MyDataset(Dataset):
def __init__(self, X, y):
self.X = X
self.y = y
def __len__(self):
return len(self.X)
def __getitem__(self, index):
return self.X[index], self.y[index]
# 定义一个全连接神经网络模型
class Net(nn.Module):
def __init__(self, input_size, hidden_size, output_size):
super(Net, self).__init__()
self.fc1 = nn.Linear(input_size, hidden_size)
self.fc2 = nn.Linear(hidden_size, output_size)
self.relu = nn.ReLU()
def forward(self, x):
x = self.fc1(x)
x = self.relu(x)
x = self.fc2(x)
return x
# 准备数据集
X_train = torch.randn(1000, 10)
y_train = torch.randint(0, 2, (1000,))
X_test = torch.randn(200, 10)
y_test = torch.randint(0, 2, (200,))
train_dataset = MyDataset(X_train, y_train)
test_dataset = MyDataset(X_test, y_test)
train_loader = DataLoader(train_dataset, batch_size=32, shuffle=True)
test_loader = DataLoader(test_dataset, batch_size=32, shuffle=False)
# 定义模型、损失函数和优化器
model = Net(10, 20, 2)
criterion = nn.CrossEntropyLoss()
optimizer = optim.SGD(model.parameters(), lr=0.01)
# 训练模型
for epoch in range(10):
running_loss = 0.0
for i, data in enumerate(train_loader, 0):
inputs, labels = data
optimizer.zero_grad()
outputs = model(inputs)
loss = criterion(outputs, labels)
loss.backward()
optimizer.step()
running_loss += loss.item()
print('Epoch %d, loss: %.3f' % (epoch+1, running_loss/len(train_loader)))
# 测试模型
y_pred = []
y_true = []
with torch.no_grad():
for data in test_loader:
inputs, labels = data
outputs = model(inputs)
_, predicted = torch.max(outputs.data, 1)
y_pred += predicted.tolist()
y_true += labels.tolist()
# 评估模型
f1 = f1_score(y_true, y_pred)
print('F1 score: %.3f' % f1)
```
在这个示例代码中,我们首先定义了一个自定义的数据集类,然后使用它来创建训练集和测试集的DataLoader。接着,我们定义了一个全连接神经网络模型、损失函数和优化器,并使用训练集数据来训练模型。最后,我们使用测试集数据来测试模型,并使用f1_score函数来评估模型的性能。
需要注意的是,这个示例代码仅仅是一个简单的示例,实际上要训练一个准确的二分类模型需要更多的数据和更复杂的模型设计。同时,我们还需要进行更多的模型调优和超参数优化才能得到更好的性能。
阅读全文