/NLP_SBERT-interpretable-semantic-text-similarity

Interpretable semantic textual similarity using SBERT.

Primary LanguageJupyter Notebook

TODOsy

  • Załadować dane
  • Skonfigurować wandb/tensorboard
  • Dataset
  • LightningDataModule
  • NLPModel 1 (ocena i typ uczone razem) [do weryfikacji]
  • NLPModel 2 (ocena i typ uczone oddzielnie)
  • NLPModel 3 (perceptron wielowarstwowy)
  • trenowanie modelu (kod)
  • eval (kod)
  • trenowanie i testowanie modelu 1 (wykonanie)
  • trenowanie i testowanie modelu 2 (wykonanie)
  • trenowanie i testowanie modelu 3 (wykonanie)

MINI taski

  • odkladanie checkpointow w trakcie treningu (tak zeby bylo mozna wracac)
  • sprawdzic jak batch_size wplywa na jakosc modelu

SBERT

https://www.sbert.net/ Model name: all-mpnet-base-v2