View Code? Open in Web Editor
NEW
this is the repository for using bert model for punctuation prediction task
Python 99.82%
Shell 0.18%
bert-punctuation-prediction's Introduction
- 🔭 My research area are natural language processing, reinforcement learning and hashtag recommendation and so on.
- 🌱 I'm currently a LLM engineer
bert-punctuation-prediction's People
Contributors
Watchers
bert-punctuation-prediction's Issues
tf.flags.DEFINE_string("glove_path", glove_path, "glove embedding path")
tf.flags.DEFINE_boolean("use_pretrained", True, "use pretrained word embedding")
from data.prepro_peopledaily import process_data
from utils import batchnize_dataset