{ "cells": [ { "cell_type": "code", "execution_count": 1, "id": "74fc32e1", "metadata": {}, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "label\n", "1 5362\n", "0 4666\n", "3 2159\n", "4 1937\n", "2 1304\n", "5 572\n", "Name: count, dtype: int64\n" ] } ], "source": [ "from datasets import load_dataset\n", "import pandas as pd\n", "\n", "dataset = load_dataset(\"dair-ai/emotion\")\n", "label_names = dataset[\"train\"].features[\"label\"].names\n", "\n", "df = pd.DataFrame(dataset[\"train\"])\n", "print(df[\"label\"].value_counts())" ] }, { "cell_type": "code", "execution_count": 2, "id": "53d69ac1", "metadata": {}, "outputs": [ { "data": { "text/html": [ "
| \n", " | text | \n", "label | \n", "
|---|---|---|
| 0 | \n", "i didnt feel humiliated | \n", "0 | \n", "
| 1 | \n", "i can go from feeling so hopeless to so damned... | \n", "0 | \n", "
| 2 | \n", "im grabbing a minute to post i feel greedy wrong | \n", "3 | \n", "
| 3 | \n", "i am ever feeling nostalgic about the fireplac... | \n", "2 | \n", "
| 4 | \n", "i am feeling grouchy | \n", "3 | \n", "
| Epoch | \n", "Training Loss | \n", "Validation Loss | \n", "Accuracy | \n", "F1 Weighted | \n", "
|---|---|---|---|---|
| 1 | \n", "0.609121 | \n", "0.200775 | \n", "0.927500 | \n", "0.928239 | \n", "
| 2 | \n", "0.169215 | \n", "0.148969 | \n", "0.940500 | \n", "0.940165 | \n", "
| 3 | \n", "0.116937 | \n", "0.145599 | \n", "0.936500 | \n", "0.936436 | \n", "
"
],
"text/plain": [
"