글 수 125

텐서플로와 케라스 코드 비교

조회 수 2645 추천 수 0 2018.06.06 20:55:46


https://gist.github.com/cburgdorf/e2fb46e5ad61ed7b9a29029c5cc30134

 

 

 

가장 유명한 딥러닝 라이브러리는 텐서플로와 케라스입니다. 간단한 XOR 문제를 학습하는 코드를 비교한 것인데 함수 사용에 있어서 큰 차이가 있음을 알 수 있습니다.

 

케라스는 model 인스턴스를 생성하고 그 위에 add로 하나씩 레이어를 쌓습니다. 그리고 compile로 오차계산(loss)과 학습방법(optimizer)을 결정하고, 마지막으로 fit을 사용해 데이터를 입력하여 학습을 시작합니다.

 

이에 반해서 텐서플로는 신경망의 가중치(w)와 편향(b) 변수를 직접 선언하고 tf.matmul로 행렬계산을 하는 코드를 작성해야 합니다. 특히 행렬의 입력과 출력의 개수를 정확히 맞춰줘야 하는데 케라스는 이런 과정을 자동으로 수행합니다.

 

 

 

확실히 케라스가 좀 더 직관적이고 사용하기가 간단합니다. 하지만 텐서플로는 구글이라는 브랜드 파워가 있고 사용자가 훨씬 많아서 예제를 찾거나 질문하기가 수월하다는 장점이 있습니다. 자신의 사용 목적에 따라서 어떤 라이브러리를 사용할 지 결정하는게 좋을 것 같습니다.

엮인글 :
List of Articles
제목 글쓴이 날짜 조회 수
YOLO, Object Detection Network - 알고리즘 설명 file 깊은바다 2019-06-02 7619
사진 한 장으로 GAN 영상 생성 깊은바다 2019-05-23 1052
객체 검출 딥러닝 모델인 YOLO의 테드 영상 깊은바다 2019-05-22 1834
스냅챕의 성별 변환 필터는 GAN을 사용했을까 깊은바다 2019-05-15 395
신경망에 활성화 함수가 필요한 이유 file 깊은바다 2019-05-14 1042
핸즈온 머신러닝 2판 영문 초고 다운로드 깊은바다 2019-05-12 1294
GTA5로 자율주행 구현 영상 깊은바다 2019-05-08 995
전신 사진을 생성하는 GAN 깊은바다 2019-05-05 519
인공신경망 학습 레시피 - 안드레이 카파시의 딥러닝 실전 노하우 깊은바다 2019-04-27 445
MS의 챗봇 린나, 딥러닝으로 노래를 부르는 가수로 데뷔 깊은바다 2019-04-10 317
10분안에 배우는 머신러닝 - GAN 알고리즘 원리와 응용분야 깊은바다 2019-04-01 763
구글 텐서플로우 첫걸음 - Tensorflow2.0 소개 [1] 깊은바다 2019-03-27 4011
스케치를 사진으로 변환하는 딥러닝 - NVIDIA의 GauGAN 깊은바다 2019-03-20 658
인공지능이 가상의 온라인 모델을 만들어낸다면? 깊은바다 2019-03-15 227
딥 러닝 이후, AI 알고리즘 트렌드 깊은바다 2019-03-11 2272