Esempio n. 1
0
def get_model():
    model = get_network()
    print('build network ......')
    print(model.summary())
    return NeuralNetworkClassifier(model,
                                   balancing_class_weight=True,
                                   filepath=MODEL_FILEPATH)
Esempio n. 2
0
def get_model():
    from model_v30 import get_network
    print('build network ......')
    model = get_network(embed_filepath=EMBED_FILEPATH)
    print(model.summary())
    return NeuralNetworkClassifier(model,
                                   balancing_class_weight=True,
                                   filepath=MODEL_FILEPATH)