Add notebook for training triplet loss model

yoheikikuta / US-patent-analysis

This is a repository of the analysis of US patent.

8 stars 8 forks source link

Add notebook for training triplet loss model #10

Closed yoheikikuta closed 5 years ago

yoheikikuta commented 5 years ago

Train a model with triplet loss. Input data is features from pre-traiend BERT. positive and negative pairs are created by using cited and not-cited info.