ESTIMATOR_TYPE_CLASSIFIER = "classifier"
ESTIMATOR_TYPE_REGRESSOR = "regressor"
ESTIMATOR_TYPE_CLUSTERER = "clusterer"

PREDICTION_KEY_PROBABILITIES = "probabilities"
PREDICTION_KEY_CLASSES = "classes"
PREDICTION_KEY_CLASS_IDS = "class_ids"
PREDICTION_KEY_CLASS_ONE_HOT_IDS = "class_one_hot_ids"
PREDICTION_KEY_ALL_CLASS_IDS = "all_class_ids"
PREDICTION_KEY_ALL_CLASSES = "all_classes"
PREDICTION_KEY_ALL_CLASSES_MAPPING = "all_classes_mapping"

from tensorflow_model_analysis import EvalConfig

ESTIMATOR_TYPE_CLASSIFIER = "classifier"
ESTIMATOR_TYPE_REGRESSOR = "regressor"
ESTIMATOR_TYPE_CLUSTERER = "clusterer"

PREDICTION_KEY_PROBABILITIES = "probabilities"
PREDICTION_KEY_CLASSES = "classes"
PREDICTION_KEY_CLASS_IDS = "class_ids"
PREDICTION_KEY_CLASS_ONE_HOT_IDS = "class_one_hot_ids"
PREDICTION_KEY_ALL_CLASS_IDS = "all_class_ids"
PREDICTION_KEY_ALL_CLASSES = "all_classes"
PREDICTION_KEY_ALL_CLASSES_MAPPING = "all_classes_mapping"

from tensorflow_model_analysis import EvalConfig
