Skip to content

Load Tensorflow pb file using Bert/TextCNNs, an ensemble model using Java.

Notifications You must be signed in to change notification settings

cyk1337/BERT-Text-Classification-J

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

8 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Ensemble Classification Demo

Implemented TextCNNs,BERT as well as the weighted vote ensemble method.

This repository contains the whole inference process of both BERT and TextCNNs, including preprocessing and prediction.

Preparation

  1. Firstly, you must have exported either BERT/TextCNNs models into .pb format. For BERT models, you can find the solution to exporting BERT .pb model by checking https://github.com/cyk1337/BERT-Classification/blob/master/bert/run_classification.py.
  2. Then, use the .pb model path to initialize the model class at
    String bert_model_path = "pb_model/bert_L4_FC3_Seq128.pb";
    , suppose you have generated .pb model correctly.

Releases

No releases published

Packages

No packages published

Languages