Skip to content

Instantly share code, notes, and snippets.

@vaskonov

vaskonov/3-2.py

Created January 30, 2019 08:08
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save vaskonov/c7acf997bdf16bb54a15e7cd54544372 to your computer and use it in GitHub Desktop.
Save vaskonov/c7acf997bdf16bb54a15e7cd54544372 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
3-2.py
from deeppavlov import configs
from deeppavlov.core.common.file import read_json
from deeppavlov import configs, train_model
model_config = read_json(configs.doc_retrieval.en_ranker_tfidf_wiki)
model_config["dataset_reader"]["data_path"] = "/content/SentenceCorpus/unlabeled_articles/plos_unlabeled"
model_config["dataset_reader"]["dataset_format"] = "txt"
ranker = train_model(model_config)
docs = ranker(['cerebellum'])
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment