Deep Learning เบื้องต้นด้วย PyTorch
Jasmin Ludolf
Senior Data Science Content Developer, DataCamp
$$







ได้ค่าระหว่าง 0 ถึง 1
ถ้า output > 0.5 หมายความว่า class label = 1 (สัตว์เลี้ยงลูกด้วยนม)
import torch import torch.nn as nn input_tensor = torch.tensor([[6]]) sigmoid = nn.Sigmoid()output = sigmoid(input_tensor) print(output)
tensor([[0.9975]])
model = nn.Sequential(
nn.Linear(6, 4), # First linear layer
nn.Linear(4, 1), # Second linear layer
nn.Sigmoid() # Sigmoid activation function
)
การใช้ Sigmoid เป็นขั้นตอนสุดท้ายใน network ของ linear layers นั้นเทียบเท่ากับ logistic regression แบบดั้งเดิม






import torch import torch.nn as nn # Create an input tensor input_tensor = torch.tensor( [[4.3, 6.1, 2.3]]) # Apply softmax along the last dimensionprobabilities = nn.Softmax(dim=-1) output_tensor = probabilities(input_tensor) print(output_tensor)
tensor([[0.1392, 0.8420, 0.0188]])
dim = -1 ระบุว่า softmax ถูกนำไปใช้กับมิติสุดท้ายของ input tensornn.Softmax() ใช้เป็นขั้นตอนสุดท้ายใน nn.Sequential() ได้Deep Learning เบื้องต้นด้วย PyTorch