구글이 skflow 에 이어서 텐서플로우의 래퍼(Wrapper) 모듈을 하나 더 발표하였습니다. 바로 프리티 텐서(Pretty Tensor) 입니다.

프리티 텐서는 메소드 체이닝 형태로 쓸 수 있는 가벼운 래퍼 라이브러리입니다. 간단한 예를 살펴 보면 금방 이해할 수 있습니다.

result = (pretty_tensor.wrap(input_data, m)
          .flatten()
          .fully_connected(200, activation_fn=tf.nn.relu)
          .fully_connected(10, activation_fn=None)
          .softmax(labels, name=softmax_name))

프리티 텐서는 skflow 와는 다르게 pypi 에 등록되어 있어 pip 를 이용해 바로 설치할 수 있습니다.

$ pip install prettytensor

사용자들이 텐서플로우를 보다 쉽게 이용할 수 있도록 구글이 많은 노력을 기울이는 것 같습니다. 보다 자세한 내용은 프리티 텐서 기트허브 페이지를 참고하세요.

https://github.com/google/prettytensor

(추가) Pretty Tensor는 0.6 버전이 되었습니다.