Skip to content
Update SDG classifier authored by Ioanna Mandilara's avatar Ioanna Mandilara
......@@ -53,7 +53,7 @@ text = ['Europe has always been the home of industry. For centuries, it has been
```python
from SDGDetector import SDGDetector
combo = SDGDetector.SDG_classifier(pretrained_model_name='XLNet',
combo = SDGDetector.SDG_classifier(pretrained_model_name='XLNet',pretrained_model_path=<the path of downloaded model>,
sentence_model_name='all-mpnet-base-v2',top_keywords=10,diversity=0.3,n_gram_range=(1,2))
sdg,sdg_name,association = combo.predict(text,return_association=True)
......
......