Introduction au Deep Learning avec PyTorch
Jasmin Ludolf
Senior Data Science Content Developer, DataCamp
$$







On obtient une valeur entre 0 et 1
Si la sortie > 0,5, étiquette = 1 (mammifère)
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), # Première couche linéaire
nn.Linear(4, 1), # Deuxième couche linéaire
nn.Sigmoid() # Fonction d'activation sigmoïde
)
La sigmoïde en dernière étape d'un réseau de couches linéaires est équivalente à la régression logistique traditionnelle






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 indique que softmax est appliquée à la dernière dimension du tenseur d'entréenn.Softmax() peut être utilisée comme dernière étape dans nn.Sequential()Introduction au Deep Learning avec PyTorch