s444415/uczenie_glebokie_projekt

Michał Kozłowski a3498e76f1 update readme with accuracy

2023-02-13 14:46:19 +01:00

722 B

Raw Blame History

Projekt na przedmiot Uczenie głębokie w przetwarzaniu tekstu

Skład grupy:

Michał Kozłowski
Szymon Jadczak

Modele:

RobertaForSequenceClassification
GPT-2
T5
FLAN-T5

Accuracy test split

RobertaForSequenceClassification -> 0.9392201834862385
GPT-2 ->
T5 -> Accuracy: 0.9129464285714286
FLAN-T5 ->

Trenowanie

Google Colab

Link do modeli na google drive

https://drive.google.com/drive/folders/1GWNah7-LZI7jrFzUpL9Le1E7b73RutbU?usp=sharing

Linki do poszczególnych modeli oraz wykresów znajdują się w notatnikach

Dataset:

Orginalny dataset: https://huggingface.co/datasets/sst2
Przetworzony dataset: https://huggingface.co/datasets/Zombely/sst2-project-dataset