레플리
글 수 203

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

조회 수 3158 추천 수 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
제목 글쓴이 날짜 조회 수
팩맨 게임을 딥러닝으로 시뮬레이션한 GameGAN 깊은바다 2020-05-23 703
CNN 시각화 사이트 - CNN Explainer 깊은바다 2020-05-10 1457
얀 르쿤, 자기지도학습이 강화학습보다 중요 깊은바다 2020-05-03 722
미디가 아니라 오디오로 음악을 만들어내는 OpenAI의 생성 모델 - Jukebox file 깊은바다 2020-05-02 661
딥러닝 웹서비스 개발 도전기 - 연예인 얼굴 판별기 깊은바다 2020-05-01 3099
사회적 거리를 지키는지 확인하는 지능형 CCTV 깊은바다 2020-04-23 477
딥페이크로 일론 머스크인 척 친구들을 속이는 영상 깊은바다 2020-04-19 1389
딥마인드의 Agent57, 아타리 게임 벤치마크 정복 깊은바다 2020-04-03 456
AutoML-Zero, 진화적 알고리즘을 사용하여 신경망 구조 설계 [2] 깊은바다 2020-03-13 820
자기지도학습 기반의 이미지 사전훈련 모델 - SimCLR 깊은바다 2020-03-10 3015
딥러닝을 사용한 포즈 인식 - Teachable Machine 깊은바다 2020-02-28 2950
딥러닝 챗봇에 대한 오해 깊은바다 2020-02-25 902
Semantic Segmentation을 활용한 차량 파손 탐지 딥러닝 모델 개발기 깊은바다 2020-02-14 734
딥러닝으로 음성합성을 해주는 서비스 - 네이버 클로바더빙 file 깊은바다 2020-02-09 471
데이팅 앱 틴더(Tinder)에서 이미지 인식으로 자동화된 봇 만들기 깊은바다 2020-02-06 662