[TFX 스터디] 0. TFX를 활용하는 프로젝트 시작

현재 TFX 스터디에서 '살아 움직이는 머신러닝 파이프라인' 책을 마치고 새로 TFX 프로젝트를 시작해 직접 머신러닝 프로젝트를 만들면서 TFX를 배우기로 했다. 아무래로 책에 나온 코드가 현재 버전과 달라진 점이 있다보니 TFX를 직접 써보면서 배우는 쪽으로 결정했다.

Machine Translator with TFX

TFX 스터디에서 만들어볼 개인 프로젝트 개요

그래서 이번에 필자가 결정한 개인 프로젝트는 기계 번역을 지속적으로 학습하고 개선하는 머신러닝 파이프라인 시스템을 만들어볼 것이다. 머신러닝 파이프라인 내에서 학습한 한국어-영어 번역 모델을 Tensorflow Serving으로 배포할 것이다. 사용자는 Gradio를 통해 모델에게 번역을 요청하고 받아볼 수 있으며, 번역 결과에 대해 피드백을 전달할 수 있도록 만들 것이다. 사용한 모델은 Seq2Seq와 Transformer 두개를 사용해볼 것이다. 머신러닝 파이프라인 내에서 모델을 변경하는 작업도 체험해보고싶기 때문이다.

앞으로 TFX 스터디에서 해당 프로젝트를 진행하면서 TFX에 대해 공부한 내용을 공유해볼 예정이다. 또한 이 프로젝트는 Github을 통해 공개할 것이다.