PyTorch: Được sử dụng rộng rãi cho nghiên cứu học sâu với cú pháp linh hoạt, dễ sử dụng
PyTorch
import torch
import torch.nn as nn
import torch.optim as optim
# Dữ liệu mẫu
X = torch.randn(100, 20)
y = torch.randint(0, 2, (100, 1)).float()
# Mô hình
class SimpleNN(nn.Module):
def __init__(self):
super(SimpleNN, self).__init__()
self.fc1 = nn.Linear(20, 64)
self.relu = nn.ReLU()
self.fc2 = nn.Linear(64, 1)
self.sigmoid = nn.Sigmoid()
def forward(self, x):
return self.sigmoid(self.fc2(self.relu(self.fc1(x))))
model = SimpleNN()
criterion = nn.BCELoss()
optimizer = optim.Adam(model.parameters(), lr=0.001)
# Huấn luyện
for epoch in range(10):
output = model(X)
loss = criterion(output, y)
optimizer.zero_grad()
loss.backward()
optimizer.step()PreviousTensorFlow: Một framework mạnh mẽ, phổ biến để xây dựng và huấn luyện mô hình học sâuNextKeras: API cấp cao chạy trên TensorFlow, hỗ trợ xây dựng mô hình nhanh chóng
Last updated
