Classes
class CosineDecay: A LearningRateSchedule that uses a cosine decay with optional warmup.
class CosineDecayRestarts: A LearningRateSchedule that uses a cosine decay schedule with restarts.
class LinearModel: Linear Model for regression and classification problems.
class SequenceFeatures: A layer for sequence input.
class WideDeepModel: Wide & Deep Model for regression and classification problems.