건조젤리의 저장소

6-2. Tensorflow를 이용한 Softmax Classifier구현 본문

공부 기록/모두를 위한 딥러닝 (Basic)

6-2. Tensorflow를 이용한 Softmax Classifier구현

건조젤리 2019. 11. 7. 14:11

김성훈 교수님의 강의내용을 정리한 내용입니다.

출처 : http://hunkim.github.io/ml/

 

모두를 위한 머신러닝/딥러닝 강의

 

hunkim.github.io

tf.nn.softmax를 이용하여 구현이 가능하다.

arg_max는 최대값의 위치를 반환한다.

더 간결하게!

Tensorflow에서 제공하는 Cross_entropy 계산

2번과 같이 Tensorflow에서 제공하는 함수를 이용할 수 있다.

만약 읽고자 하는 데이터의 Y값이 One-Hot 형식으로 구성되지 않았다면

tf.one_hot을 이용하여 바꿀 수 있다.

주의할 점은 이 함수를 이용하면 출력 차원이 1개 늘어나게 되는데 tf.reshape를 이용하여 차원을 다시 바꿔주자.

 

'공부 기록 > 모두를 위한 딥러닝 (Basic)' 카테고리의 다른 글

7-2. Learning and test data sets  (0) 2019.11.07
7-1. Learning rate, data preprocessing, overfitting  (0) 2019.11.07
6-1. Softmax classification: Multinomial classification  (0) 2019.11.07
5-2. Tensorflow를 이용한 Logistic (regression) classifier구현  (0) 2019.11.07
5-1. Logistic (regression) classification  (0) 2019.11.07
'공부 기록/모두를 위한 딥러닝 (Basic)' Related Articles more
Comments