Bram van Es commited on
Commit
797ad44
·
1 Parent(s): 5d8e89e

added models

Browse files
Files changed (1) hide show
  1. config.py +8 -13
config.py CHANGED
@@ -16,8 +16,10 @@ DEFAULT_MODELS = {
16
  "roberta-base",
17
  "albert-base-v2"
18
  "UMCU/CardioMedRoBERTa.nl",
19
- "UMCU/CardioBERTa.nl",
20
  "UMCU/CardioBERTa.nl_clinical",
 
 
21
  "CLTL/MedRoBERTa.nl",
22
  "DTAI-KULeuven/robbert-2023-dutch-base",
23
  "DTAI-KULeuven/robbert-2023-dutch-large"
@@ -53,8 +55,6 @@ VIZ_SETTINGS = {
53
 
54
  # Processing settings
55
  PROCESSING_SETTINGS = {
56
- "default_iterations": 1,
57
- "max_iterations": 10,
58
  "epsilon": 1e-10 # Small value to avoid log(0)
59
  }
60
 
@@ -73,32 +73,27 @@ UI_SETTINGS = {
73
  {
74
  "text": "The quick brown fox jumps over the lazy dog.",
75
  "model": "gpt2",
76
- "type": "decoder",
77
- "iterations": 1
78
  },
79
  {
80
  "text": "The capital of France is Paris.",
81
  "model": "bert-base-uncased",
82
- "type": "encoder",
83
- "iterations": 1
84
  },
85
  {
86
  "text": "Quantum entanglement defies classical physics intuition completely.",
87
  "model": "distilgpt2",
88
- "type": "decoder",
89
- "iterations": 1
90
  },
91
  {
92
  "text": "Buffalo buffalo Buffalo buffalo buffalo buffalo Buffalo buffalo.",
93
  "model": "gpt2",
94
- "type": "decoder",
95
- "iterations": 1
96
  },
97
  {
98
  "text": "Machine learning algorithms require computational resources.",
99
  "model": "distilbert-base-uncased",
100
- "type": "encoder",
101
- "iterations": 1
102
  }
103
  ]
104
  }
 
16
  "roberta-base",
17
  "albert-base-v2"
18
  "UMCU/CardioMedRoBERTa.nl",
19
+ "UMCU/CardioBERTa_base.nl",
20
  "UMCU/CardioBERTa.nl_clinical",
21
+ "UMCU/CardioDeBERTa.nl",
22
+ "UMCU/CardioDeBERTa.nl_clinical",
23
  "CLTL/MedRoBERTa.nl",
24
  "DTAI-KULeuven/robbert-2023-dutch-base",
25
  "DTAI-KULeuven/robbert-2023-dutch-large"
 
55
 
56
  # Processing settings
57
  PROCESSING_SETTINGS = {
 
 
58
  "epsilon": 1e-10 # Small value to avoid log(0)
59
  }
60
 
 
73
  {
74
  "text": "The quick brown fox jumps over the lazy dog.",
75
  "model": "gpt2",
76
+ "type": "decoder"
 
77
  },
78
  {
79
  "text": "The capital of France is Paris.",
80
  "model": "bert-base-uncased",
81
+ "type": "encoder"
 
82
  },
83
  {
84
  "text": "Quantum entanglement defies classical physics intuition completely.",
85
  "model": "distilgpt2",
86
+ "type": "decoder"
 
87
  },
88
  {
89
  "text": "Buffalo buffalo Buffalo buffalo buffalo buffalo Buffalo buffalo.",
90
  "model": "gpt2",
91
+ "type": "decoder"
 
92
  },
93
  {
94
  "text": "Machine learning algorithms require computational resources.",
95
  "model": "distilbert-base-uncased",
96
+ "type": "encoder"
 
97
  }
98
  ]
99
  }