model_function_basico.py

import tensorflow as tf

def model_fn(features, labels, mode):
   # Deve conter os seguintes passos:
   # 1. Definir o modelo via TF
   # 2. Definir como gerar predições
   # 3. Definir a loss function para treino e avaliação
   # 4. Definir como otimizar o modelo (optimizer)
   # 5. Definir métricas para avaliação
   # 6. Retornar um EstimatorSpec definindo os passos acima
   return tf.estimator.EstimatorSpec(mode, predictions, loss, train_op, eval_metric_ops)