Views
No views yet
pip install -U sentence-transformers1from sentence_transformers import SentenceTransformer
2latex = r"13\times x"
3pmml = r"<math><semantics><mrow><mn>13</mn><mo>×</mo><mi>x</mi></mrow></semantics></math>"
4cmml = r"<math><apply><times></times><cn>13</cn><ci>x</ci></apply></math>"
5
6model = SentenceTransformer('Jyiyiyiyi/CLFE_ConMath')
7
8embedding_latex = model.encode([{'latex': latex}])
9embedding_pmml = model.encode([{'mathml': pmml}])
10embedding_cmml = model.encode([{'mathml': cmml}])
11
12print('latex embedding:')
13print(embedding_latex)
14print('Presentation MathML embedding:')
15print(embedding_pmml)
16print('Content MathML embedding:')
17print(embedding_cmml)SentenceTransformer(
(0): Asym(
(latex-0): Transformer({'max_seq_length': 512, 'do_lower_case': False}) with Transformer model: MPNetModel
(mathml-0): MarkuplmTransformer({'max_seq_length': 512, 'do_lower_case': False}) with Transformer model: MarkupLMModel
)
(1): Pooling({'word_embedding_dimension': 768, 'pooling_mode_cls_token': False, 'pooling_mode_mean_tokens': True, 'pooling_mode_max_tokens': False, 'pooling_mode_mean_sqrt_len_tokens': False, 'pooling_mode_weightedmean_tokens': False, 'pooling_mode_lasttoken': False})
)@inproceedings{wang2023math,
title={Math Information Retrieval with Contrastive Learning of Formula Embeddings},
author={Wang, Jingyi and Tian, Xuedong},
booktitle={International Conference on Web Information Systems Engineering},
pages={97--107},
year={2023},
organization={Springer}
}