레플리
글 수 203

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

조회 수 3156 추천 수 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
제목 글쓴이 날짜 조회 수
오픈소스로 공개된 Text-to-Image 모델 - Stable Diffusion 깊은바다 2022-09-07 1336
Dall.E 2 Prompt Book - 프롬프트 작성 가이드 깊은바다 2022-07-14 4339
최초로 인간 수준의 크기를 가진 중국의 딥러닝 모델 - BaGuaLu file 깊은바다 2022-06-23 1116
미국에서 Dalle Mini로 짤(Meme)을 만드는게 유행 file 깊은바다 2022-06-19 1582
말로 영상을 생성하는 딥러닝 모델 - CogVideo file 깊은바다 2022-05-30 1069
DALL.E를 직접 테스트해볼 수 있는 사이트 - DALL.E mini file 깊은바다 2022-05-17 11518
전문가 수준의 사진을 생성하는 NVIDIA의 PoE-GAN 깊은바다 2022-02-16 1279
메타AI의 멀티모달 자기지도학습 모델인 Data2vec 깊은바다 2022-01-26 1028
GAN으로 사진에서 특정 부분을 지워주는 서비스 - Cleanup.pictures file 깊은바다 2022-01-17 1018
3D 이미지를 만드는 GAN - StyleNeRF 깊은바다 2021-12-20 835
CLIP을 사용하여 말로 그림을 그려주는 스마트폰 앱 - WOMBO Dream file 깊은바다 2021-11-25 1777
내가 말하는 걸 보여줘, DALL-E 깊은바다 2021-11-11 1062
텍스트를 이미지로 생성해주는 사이트 - Text2Art (VQGAN-CLIP) file 깊은바다 2021-11-10 6865
원티드는 어떻게 AI로 채용 결과를 예측하고 매칭할까 깊은바다 2021-11-09 959
OpenAI의 CLIP을 한글로 구현한 KoCLIP file 깊은바다 2021-11-02 2732