def get_classificationxlnet_model(model_config,
                                  run_config,
                                  n_class,
                                  summary_type="last"):
    model = modeling.ClassificationXLNetModel(model_config,
                                              run_config,
                                              n_class,
                                              summary_type,
                                              name="model")
    return model
예제 #2
0
def get_classificationxlnet_model(model_config, run_config, n_class):
    model = modeling.ClassificationXLNetModel(model_config,
                                              run_config,
                                              n_class,
                                              name="model")
    return model