Responder a: Como carregar modelo treinado?

#37864
Dalton Vargas
Moderador

Olá Renato!

Certo, a ideia é essa mesmo que você mencionou, faz um load_model. Segue abaixo um exemplo de função que você poderá usar para a maioria dos modelos criados aqui no curso:

# Vetor para indicar qual modelo iremos carregar
MODEL_TYPES = ['SOM', 'EMOÇÃO']

# Função para carregar o modelo
def load_model_by_name(model_type):
    if model_type == MODEL_TYPES[0]:
       model = tf.keras.models.load_model('models/ambient_sound_classification.hdf5')
       model_dict = sorted(list(['ar_condicionado', 'buzina_de_carro', 'crianca_brincando', 'latido_de_cachorro', 'perfuracao', 'motor_em_marcha_lenta', 'tiro_de_arma', 'britadeira', 'sirene', 'musica_de_rua']))
       SAMPLE_RATE = 44100 # taxa de amostragem do modelo criado

    return model, model_dict, SAMPLE_RATE