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

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

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

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

---

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

---

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

---

더 간결하게!

Tensorflow에서 제공하는 Cross_entropy 계산

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

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

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

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