Missed TensorFlow World? Check out the recap. Learn more

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