from keras.feature_column.sequence_feature_column import SequenceFeatures as SequenceFeatures
from keras.optimizers.schedules.learning_rate_schedule import CosineDecay as CosineDecay, CosineDecayRestarts as CosineDecayRestarts
from keras.premade_models.linear import LinearModel as LinearModel
from keras.premade_models.wide_deep import WideDeepModel as WideDeepModel