[머신러닝] 9. 머신러닝 라이브러리 비교

머신러닝 라이브러리

머신러닝을 공부하기 위해 지원하는 라이브러리들이 여러가지가 있다.

대표적인 라이브러리들을 알아보자.

  • 라이브러리
    • TensorFlow (구글이 만든 딥러닝[인공신경망] 라이브러리)
    • Scikit-Learn
    • PyTorch
    • Keras
    • Caffe2
    • theano

라이브러리 비교

  텐서플로우 사이킷런 파이토치 케라스
장점 텐서보드를 통해서 파라미터 변화 양상이나 DNN구조를 알 수 있습니다. 탄탄한 학습 알고리즘을 가지고 있습니다. 간단하고 직관적으로 학습 가능 배우기 쉽고 모델 구축하기 쉽다.
단점 메모리 효율적사용X 딥러닝 & 강화학습 다루지않음 텐서플로우에 비해 사용자 얕고 자료 많이 없음 오류가 발생할 경우 케라스문제인지 백엔드 문제인지 알 수 없음

머신러닝 라이브러리로는 ‘사이킷-런’

딥러닝 라이브러리로는 ‘텐서플로우’