Identifying topics by training NTM in SageMaker