diff --git a/README.md b/README.md index bd2acd0..a9ca8f3 100644 --- a/README.md +++ b/README.md @@ -19,6 +19,55 @@ task_categories: task_ids: - sentiment-classification paperswithcode_id: imdb-movie-reviews +train-eval-index: +- config: plain_text + task: text-classification + task_id: binary_classification + splits: + train_split: train + eval_split: test + col_mapping: + text: text + label: target + metrics: + - type: accuracy + - name: Accuracy + - type: f1 + name: F1 macro + args: + average: macro + - type: f1 + name: F1 micro + args: + average: micro + - type: f1 + name: F1 weighted + args: + average: weighted + - type: precision + name: Precision macro + args: + average: macro + - type: precision + name: Precision micro + args: + average: micro + - type: precision + name: Precision weighted + args: + average: weighted + - type: recall + name: Recall macro + args: + average: macro + - type: recall + name: Recall micro + args: + average: micro + - type: recall + name: Recall weighted + args: + average: weighted --- # Dataset Card for "imdb"