모두의 연구소 AIFFEL 교육

전체적인 model.fit() 과정

robotgo 2021. 2. 1. 12:00

1. x,y 데이터셋 설정

2. 0,1 클래스 설정

3. 0.0003 정도의 learning rate 설정

4. 가중치 설정

5. 바이어스 설정

6. log-odds 값  구하기

7. log-odds + softmax 값 구하기

8. cross_entropy로

최종 loss-function 구하기

9. optimizer 안에 러닝 레이트, 로스 넣기

10. argmax로 acciracy 구하기

11. train data, test data 라벨링

12. optimizer, 트레인 데이터 넣고 학습

13. test data의 loss, accuracy 출력