| from bert_serving.client import BertClient | |
| # make a connection with the BERT server using it's ip address; do not give any ip if same computer | |
| bc = BertClient(ip="SERVER_IP_HERE") | |
| # get the embedding | |
| embedding = bc.encode(["I love data science and analytics vidhya."]) | |
| # check the shape of embedding, it should be 1x768 | |
| print(embedding.shape) |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment