TF 2.0 is out! Get hands-on practice at TF World, Oct 28-31. Use code TF20 for 20% off select passes. Register now

tf.contrib.learn.ProblemType

View source on GitHub

Class ProblemType

Enum-like values for the type of problem that the model solves.

THIS CLASS IS DEPRECATED.

These values are used when exporting the model to produce the appropriate signature function for serving.

The following values are supported: UNSPECIFIED: Produces a predict signature_fn. CLASSIFICATION: Produces a classify signature_fn. LINEAR_REGRESSION: Produces a regression signature_fn. LOGISTIC_REGRESSION: Produces a classify signature_fn.

Class Members

  • CLASSIFICATION = 1
  • LINEAR_REGRESSION = 2
  • LOGISTIC_REGRESSION = 3
  • UNSPECIFIED = 0