AutoModel

class tfts.models.auto_model.AutoModel(model, config)

Bases: BaseModel

tfts auto model input tensor: [batch_size, sequence_length, num_features] output tensor: [batch_size, predict_sequence_length, num_labels]

Inherited-members:

Methods

build_model(inputs)
compute_output_shape(input_shape)
from_config(config[, predict_sequence_length])
from_pretrained(weights_dir[, ...])
get_config()
load_pretrained_weights(weights_dir)
predict(x, **kwargs)
save_model(weights_dir)
save_pretrained(save_directory[, ...])
save_weights(weights_path)
summary()
to_model()