BecomeAllan
commited on
Commit
·
cf2df3b
1
Parent(s):
4bb8faf
label map
Browse files- Util_funs.py +10 -1
Util_funs.py
CHANGED
|
@@ -769,7 +769,16 @@ patterns = {
|
|
| 769 |
|
| 770 |
patterns = {x.lower():y for x,y in patterns.items()}
|
| 771 |
|
| 772 |
-
LABEL_MAP = {'negative': 0,
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 773 |
|
| 774 |
class SLR_DataSet(Dataset):
|
| 775 |
def __init__(self, **args):
|
|
|
|
| 769 |
|
| 770 |
patterns = {x.lower():y for x,y in patterns.items()}
|
| 771 |
|
| 772 |
+
LABEL_MAP = {'negative': 0,
|
| 773 |
+
'not included':0,
|
| 774 |
+
'0':0,
|
| 775 |
+
0:0,
|
| 776 |
+
'excluded':0,
|
| 777 |
+
'positive': 1,
|
| 778 |
+
'included':1,
|
| 779 |
+
'1':1,
|
| 780 |
+
1:1,
|
| 781 |
+
}
|
| 782 |
|
| 783 |
class SLR_DataSet(Dataset):
|
| 784 |
def __init__(self, **args):
|