src.pages.annotate.model

Functions

get_model(num_outputs, input_shape)
create_model_mil(shape, units[, learning_rate])
train_model(model, x_train, y_train)

Module Contents

src.pages.annotate.model.get_model(num_outputs, input_shape)

src.pages.annotate.model.create_model_mil(shape, units, learning_rate=0.01)

src.pages.annotate.model.train_model(model, x_train, y_train)