https://github.com/gameofdimension/java-bert-predict

turn bert pretrain checkpoint into saved model for a feature extracting demo in java
https://github.com/gameofdimension/java-bert-predict

bert feature-extraction google-bert java tensorflow

Last synced: about 1 month ago
JSON representation

turn bert pretrain checkpoint into saved model for a feature extracting demo in java

• Host: GitHub
• URL: https://github.com/gameofdimension/java-bert-predict
• Owner: gameofdimension
• Created: 2018-11-25T03:49:02.000Z (about 6 years ago)
• Default Branch: master
• Last Pushed: 2018-11-25T04:12:45.000Z (about 6 years ago)
• Last Synced: 2024-08-02T08:09:50.935Z (4 months ago)
• Topics: bert, feature-extraction, google-bert, java, tensorflow
• Language: Java
• Homepage:
• Size: 3.91 KB
• Stars: 33
• Watchers: 2
• Forks: 13
• Open Issues: 5
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

• awesome-bert - gameofdimension/java-bert-predict

README

        
# java-bert-predict

turn bert pretrain checkpoint into saved model for a feature extracting demo in java

# usage

1. download google bert pretrain from [here](https://storage.googleapis.com/bert_models/2018_11_03/multilingual_L-12_H-768_A-12.zip) and unzip it into a proper path.

2. because java cannot work with tensorflow checkpoint directly, we need to transform checkpoint into saved_model with following command 

	> python script/checkpoint_to_saved_model.py path/to/unzipped/pretrain/checkpoint path/to/save/model

1. run java program with path/to/save/model as params