updated links from sgnlp to sgnlp-models
Browse files
README.md
CHANGED
|
@@ -69,10 +69,10 @@ from sgnlp.models.lsr import LsrModel, LsrConfig, LsrPreprocessor, LsrPostproces
|
|
| 69 |
from transformers import cached_path
|
| 70 |
|
| 71 |
# Download files from azure blob storage
|
| 72 |
-
rel2id_path = cached_path('https://storage.googleapis.com/sgnlp/models/lsr/rel2id.json')
|
| 73 |
-
word2id_path = cached_path('https://storage.googleapis.com/sgnlp/models/lsr/word2id.json')
|
| 74 |
-
ner2id_path = cached_path('https://storage.googleapis.com/sgnlp/models/lsr/ner2id.json')
|
| 75 |
-
rel_info_path = cached_path('https://storage.googleapis.com/sgnlp/models/lsr/rel_info.json')
|
| 76 |
|
| 77 |
PRED_THRESHOLD = 0.3
|
| 78 |
preprocessor = LsrPreprocessor(rel2id_path=rel2id_path, word2id_path=word2id_path, ner2id_path=ner2id_path)
|
|
@@ -80,8 +80,8 @@ postprocessor = LsrPostprocessor.from_file_paths(rel2id_path=rel2id_path, rel_in
|
|
| 80 |
pred_threshold=PRED_THRESHOLD)
|
| 81 |
|
| 82 |
# Load model
|
| 83 |
-
config = LsrConfig.from_pretrained('https://storage.googleapis.com/sgnlp/models/lsr/v2/config.json')
|
| 84 |
-
model = LsrModel.from_pretrained('https://storage.googleapis.com/sgnlp/models/lsr/v2/pytorch_model.bin', config=config)
|
| 85 |
model.eval()
|
| 86 |
|
| 87 |
# DocRED-like instance
|
|
@@ -174,8 +174,8 @@ Please contact the authors to get the dataset if you have a valid LDC license.
|
|
| 174 |
- **Training Config:** Not available.
|
| 175 |
|
| 176 |
# Model Parameters
|
| 177 |
-
- **Model Weights:** [link](https://storage.googleapis.com/sgnlp/models/lsr/pytorch_model.bin)
|
| 178 |
-
- **Model Config:** [link](https://storage.googleapis.com/sgnlp/models/lsr/config.json)
|
| 179 |
- **Model Inputs:** Coreference clusters of entities, relations between clusters of entities, and text.
|
| 180 |
- **Model Outputs:** Scores of all possible relation labels between all possible pairs of entity clusters.
|
| 181 |
- **Model Size:** ~85MB
|
|
|
|
| 69 |
from transformers import cached_path
|
| 70 |
|
| 71 |
# Download files from azure blob storage
|
| 72 |
+
rel2id_path = cached_path('https://storage.googleapis.com/sgnlp-models/models/lsr/rel2id.json')
|
| 73 |
+
word2id_path = cached_path('https://storage.googleapis.com/sgnlp-models/models/lsr/word2id.json')
|
| 74 |
+
ner2id_path = cached_path('https://storage.googleapis.com/sgnlp-models/models/lsr/ner2id.json')
|
| 75 |
+
rel_info_path = cached_path('https://storage.googleapis.com/sgnlp-models/models/lsr/rel_info.json')
|
| 76 |
|
| 77 |
PRED_THRESHOLD = 0.3
|
| 78 |
preprocessor = LsrPreprocessor(rel2id_path=rel2id_path, word2id_path=word2id_path, ner2id_path=ner2id_path)
|
|
|
|
| 80 |
pred_threshold=PRED_THRESHOLD)
|
| 81 |
|
| 82 |
# Load model
|
| 83 |
+
config = LsrConfig.from_pretrained('https://storage.googleapis.com/sgnlp-models/models/lsr/v2/config.json')
|
| 84 |
+
model = LsrModel.from_pretrained('https://storage.googleapis.com/sgnlp-models/models/lsr/v2/pytorch_model.bin', config=config)
|
| 85 |
model.eval()
|
| 86 |
|
| 87 |
# DocRED-like instance
|
|
|
|
| 174 |
- **Training Config:** Not available.
|
| 175 |
|
| 176 |
# Model Parameters
|
| 177 |
+
- **Model Weights:** [link](https://storage.googleapis.com/sgnlp-models/models/lsr/pytorch_model.bin)
|
| 178 |
+
- **Model Config:** [link](https://storage.googleapis.com/sgnlp-models/models/lsr/config.json)
|
| 179 |
- **Model Inputs:** Coreference clusters of entities, relations between clusters of entities, and text.
|
| 180 |
- **Model Outputs:** Scores of all possible relation labels between all possible pairs of entity clusters.
|
| 181 |
- **Model Size:** ~85MB
|