|
|
from dataclasses import dataclass |
|
|
from enum import Enum |
|
|
|
|
|
@dataclass |
|
|
class Task: |
|
|
benchmark: str |
|
|
metric: str |
|
|
col_name: str |
|
|
category: str = None |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class Tasks(Enum): |
|
|
|
|
|
|
|
|
task1 = Task("NetQA", "rouge1", "NetQA_rouge1") |
|
|
task2 = Task("NetQA", "rouge2", "NetQA_rouge2") |
|
|
task3 = Task("NetQA", "rougeL", "NetQA_rougeL") |
|
|
task4 = Task("NetQA", "bert_score_f1", "NetQA_bert_score_f1") |
|
|
|
|
|
task5 = Task("Embed", "rouge1", "Embed_rouge1") |
|
|
task6 = Task("Embed", "rouge2", "Embed_rouge2") |
|
|
task7 = Task("Embed", "rougeL", "Embed_rougeL") |
|
|
task8 = Task("Embed", "bert_score_f1", "Embed_bert_score_f1") |
|
|
|
|
|
task9 = Task("Metric", "acc", "Metric_acc") |
|
|
task10 = Task("Metric", "f1", "Metric_f1") |
|
|
|
|
|
task11 = Task("CodeQA", "codebert_score_precision", "CodeQA_codebert_score_precision") |
|
|
task12 = Task("CodeQA", "codebert_score_recall", "CodeQA_codebert_score_recall") |
|
|
task13 = Task("CodeQA", "codebert_score_f1", "CodeQA_codebert_score_f1") |
|
|
task14 = Task("CodeQA", "codebert_score_f3", "CodeQA_codebert_score_f3") |
|
|
|
|
|
task15 = Task("Corpus", "acc", "Corpus_acc") |
|
|
task16 = Task("Corpus", "f1", "Corpus_f1") |
|
|
|
|
|
task17 = Task("CDTier", "acc", "CDTier_acc") |
|
|
task18 = Task("CDTier", "f1", "CDTier_f1") |
|
|
|
|
|
task19 = Task("NER", "entity_f1", "NER_entity_f1") |
|
|
|
|
|
task20 = Task("HackerNews", "rouge1", "HackerNews_rouge1") |
|
|
task21 = Task("HackerNews", "rouge2", "HackerNews_rouge2") |
|
|
task22 = Task("HackerNews", "rougeL", "HackerNews_rougeL") |
|
|
task23 = Task("HackerNews", "bert_score_f1", "HackerNews_bert_score_f1") |
|
|
|
|
|
task24 = Task("MaliURLs", "acc", "MaliURLs_acc") |
|
|
task25 = Task("MaliURLs", "f1", "MaliURLs_f1") |
|
|
|
|
|
task26 = Task("CSIC2010", "acc", "CSIC2010_acc") |
|
|
task27 = Task("CSIC2010", "f1", "CSIC2010_f1") |
|
|
|
|
|
task28 = Task("BETH", "acc", "BETH_acc") |
|
|
task29 = Task("BETH", "f1", "BETH_f1") |
|
|
|
|
|
task30 = Task("MITRE", "acc", "MITRE_acc") |
|
|
task31 = Task("MITRE", "f1", "MITRE_f1") |
|
|
|
|
|
|
|
|
task32 = Task("TASK_AVG", "CyberKUT", "CyberKUT ⬆️") |
|
|
task33 = Task("TASK_AVG", "CyberNLP", "CyberNLP ⬆️") |
|
|
task34 = Task("TASK_AVG", "CyberDSA", "CyberDSA ⬆️") |
|
|
task35 = Task("TASK_AVG", "CLS", "CLS ⬆️") |
|
|
task36 = Task("TASK_AVG", "GEN", "GEN ⬆️") |
|
|
task37 = Task("TASK_AVG", "REA", "REA ⬆️") |
|
|
|
|
|
task38 = Task("MODEL_AVG", "overall", "Overall ⬆️") |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
NUM_FEWSHOT = 0 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
TITLE = """<h1 align="center" id="space-title">SecBen</h1>""" |
|
|
|
|
|
|
|
|
INTRODUCTION_TEXT = """ |
|
|
INTRO TEXT TEST |
|
|
""" |
|
|
|
|
|
|
|
|
LLM_BENCHMARKS_TEXT = f""" |
|
|
None |
|
|
""" |
|
|
|
|
|
EVALUATION_QUEUE_TEXT = """ |
|
|
## Some good practices before submitting a model |
|
|
|
|
|
### 1) Make sure you can load your model and tokenizer using AutoClasses: |
|
|
```python |
|
|
from transformers import AutoConfig, AutoModel, AutoTokenizer |
|
|
config = AutoConfig.from_pretrained("your model name", revision=revision) |
|
|
model = AutoModel.from_pretrained("your model name", revision=revision) |
|
|
tokenizer = AutoTokenizer.from_pretrained("your model name", revision=revision) |
|
|
``` |
|
|
If this step fails, follow the error messages to debug your model before submitting it. It's likely your model has been improperly uploaded. |
|
|
|
|
|
Note: make sure your model is public! |
|
|
Note: if your model needs `use_remote_code=True`, we do not support this option yet but we are working on adding it, stay posted! |
|
|
|
|
|
### 2) Convert your model weights to [safetensors](https://huggingface.co/docs/safetensors/index) |
|
|
It's a new format for storing weights which is safer and faster to load and use. It will also allow us to add the number of parameters of your model to the `Extended Viewer`! |
|
|
|
|
|
### 3) Make sure your model has an open license! |
|
|
This is a leaderboard for Open LLMs, and we'd love for as many people as possible to know they can use your model 🤗 |
|
|
|
|
|
### 4) Fill up your model card |
|
|
When we add extra information about models to the leaderboard, it will be automatically taken from the model card |
|
|
|
|
|
## In case of model failure |
|
|
If your model is displayed in the `FAILED` category, its execution stopped. |
|
|
Make sure you have followed the above steps first. |
|
|
If everything is done, check you can launch the EleutherAIHarness on your model locally, using the above command without modifications (you can add `--limit` to limit the number of examples per task). |
|
|
""" |
|
|
|
|
|
CITATION_BUTTON_LABEL = "Copy the following snippet to cite these results" |
|
|
CITATION_BUTTON_TEXT = r""" |
|
|
""" |
|
|
|