Views
No views yet
1>>> from sentence_transformers.cross_encoder import CrossEncoder
2>>> model = CrossEncoder(model_save_path, device="cuda", max_length=64)
3>>> sentences = ["今天天气不错", "今天心情不错"]
4>>> score = model.predict([sentences])
5>>> print(score[0])