garage.tf.regressors.categorical_mlp_regressor_model module