MultiEL
Multilingual Entity Linking model by BELA model
This project want to create easy-to-use Multilingual Entity Linking model by BELA model.
Origin Project
- Bi-encoder Entity Linking Architecture (BELA): https://github.com/facebookresearch/BELA
Install
pip install multiel
Usage
from multiel import BELA
bela_run = BELA(device="cuda")
print(bela_run.process_batch(["นายกประยุทธ์ประกาศจัดการเลือกตั้ง"]))
output:
[{'offsets': [0], 'lengths': [12], 'entities': ['Q2108126'], 'md_scores': [0.22365164756774902], 'el_scores': [0.6967974901199341]}]
API
from multiel import BELA
BELA(
md_threshold:float=0.2,
el_threshold:float=0.4,
checkpoint_name: str="wiki",
device: str="cuda:0",
config_name:str="joint_el_mel_new",
repo:str="wannaphong/BELA"
)
- md_threshold: md threshold
- el_threshold: Entity Linking threshold
- checkpoint_name: checkpoint name (wiki, aida, mewsli, and e2e) or your file name with extension
- device: device
- config_name: config name (in the BELA project)
- repo: Huggingface Hub repo (Default wannaphong/BELA)
Predict
BELA.process_batch([str, str])
License
MIT license and the model is MIT license. (BELA is MIT licensed)
1 maintainer
Release files for MultiEL 0.5
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Source distribution (sdist)
| File | Size | Uploaded | |
|---|---|---|---|
| MultiEL-0.5.tar.gz | 4.1 MB | Details |
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| MultiEL-0.5-py3-none-any.whl | Python 3 | none | any | Details |
Total release size: 8.4 MB
Release files / MultiEL-0.5.tar.gz
| Download URL | MultiEL-0.5.tar.gz |
|---|---|
| Size | 4.1 MB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
0aee992a50cb1632323bc327288f4f6c14e444f12cb7af4dd5f1790641e82718
|
|
BLAKE2b-256 checksum How to use checksums |
497947ab743a239de6ff22795b884aff8be8e1029aa9b66aea58d538420e20af
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/4.0.1 CPython/3.11.4
|
Release files / MultiEL-0.5-py3-none-any.whl
| Download URL | MultiEL-0.5-py3-none-any.whl |
|---|---|
| Size | 4.2 MB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
c036e85f7f048339bdcdcf8401beca484351bbf9440d80cbf5ac8000ec465ec2
|
|
BLAKE2b-256 checksum How to use checksums |
9a5753c07bf769067c8ebdb352d356fa75040aa7a91226aca4504e6c07ad9767
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/4.0.1 CPython/3.11.4
|
