데이터 취업 스쿨 스터디 노트 -(71) pytorch

설치

임포트

import torch

pytorch 사용

일반적인 파이썬 코드

requires_grad=True -> 기울기를 계산할 수 있게 해줘라.

기울기 계산

y.backward()

chain rule

z -> x -> a : 2 2 = 4
z -> y -> a : 3 10a = 30a
=> a= 2.0이라고 했으니 4 + 30(2.0) = 64

보스턴 집값예측

nn: 뉴럴네트워크

optimizer.zero_grad(): 미분치를 0으로 세팅을 다시.(가중치를 업데이트를 하기위한 재료로 쓰는 gradient를 0으로 세팅)

유방암 예측

이진분류이니까 출력을 시그모이드로