swardiantara commited on
Commit
a3126f7
·
verified ·
1 Parent(s): 9f34b2e

Upload 2 files

Browse files

update readme.md and attach task's illustration

Files changed (2) hide show
  1. README.md +108 -3
  2. concept.png +0 -0
README.md CHANGED
@@ -1,3 +1,108 @@
1
- ---
2
- license: mit
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ pipeline_tag: token-classification
3
+ tags:
4
+ - drone-forensics
5
+ - event-recognition
6
+ license: mit
7
+ language:
8
+ - en
9
+ base_model:
10
+ - FacebookAI/roberta-base
11
+ library_name: transformers
12
+ ---
13
+
14
+ # ADFLER-roberta-base
15
+
16
+ This is a [roberta-base](https://huggingface.co/FacebookAI/roberta-base) model fine-tuned on a collection of drone flight log messages: It performs log event recognition by assigning NER tag to each token within the input message using the BIOES tagging scheme.
17
+
18
+ For more detailed information about the model, please refer to the Roberta's model card.
19
+
20
+ <!--- Describe your model here -->
21
+
22
+ ## Intended Use
23
+ ![Description of Image](./concept.png)
24
+ - Use to split log records into sentences as well as detecting if the sentence is an event message or not.
25
+ - This model is trained diverse drone log messages from various models acquired from [Air Data](https://app.airdata.com/wiki/Notifications/)
26
+ ## Usage (Transformers)
27
+
28
+ Using this model becomes easy when you have [transformers](https://www.SBERT.net) installed:
29
+
30
+ ```
31
+ pip install -U transformers
32
+ ```
33
+
34
+ Then you can use the model like this:
35
+
36
+ ```python
37
+ >>> from transformers import pipeline
38
+ >>> model = pipeline('ner', model='swardiantara/ADFLER-roberta-base')
39
+
40
+ >>> model("Unknown Error, Cannot Takeoff. Contact DJI support.")
41
+
42
+ [{'entity': 'B-Event',
43
+ 'score': np.float32(0.9991462),
44
+ 'index': 1,
45
+ 'word': 'Unknown',
46
+ 'start': 0,
47
+ 'end': 7},
48
+ {'entity': 'E-Event',
49
+ 'score': np.float32(0.9971226),
50
+ 'index': 2,
51
+ 'word': 'ĠError',
52
+ 'start': 8,
53
+ 'end': 13},
54
+ {'entity': 'B-Event',
55
+ 'score': np.float32(0.9658275),
56
+ 'index': 4,
57
+ 'word': 'ĠCannot',
58
+ 'start': 15,
59
+ 'end': 21},
60
+ {'entity': 'E-Event',
61
+ 'score': np.float32(0.9913662),
62
+ 'index': 5,
63
+ 'word': 'ĠTake',
64
+ 'start': 22,
65
+ 'end': 26},
66
+ {'entity': 'E-Event',
67
+ 'score': np.float32(0.9961124),
68
+ 'index': 6,
69
+ 'word': 'off',
70
+ 'start': 26,
71
+ 'end': 29},
72
+ {'entity': 'B-NonEvent',
73
+ 'score': np.float32(0.9994654),
74
+ 'index': 8,
75
+ 'word': 'ĠContact',
76
+ 'start': 31,
77
+ 'end': 38},
78
+ {'entity': 'I-NonEvent',
79
+ 'score': np.float32(0.9946643),
80
+ 'index': 9,
81
+ 'word': 'ĠDJ',
82
+ 'start': 39,
83
+ 'end': 41},
84
+ {'entity': 'I-NonEvent',
85
+ 'score': np.float32(0.8926663),
86
+ 'index': 10,
87
+ 'word': 'I',
88
+ 'start': 41,
89
+ 'end': 42},
90
+ {'entity': 'E-NonEvent',
91
+ 'score': np.float32(0.9982748),
92
+ 'index': 11,
93
+ 'word': 'Ġsupport',
94
+ 'start': 43,
95
+ 'end': 50}]
96
+ ```
97
+
98
+ ## Citing & Authors
99
+ ```bibtex
100
+ @misc{albert_ner_model,
101
+ author={Silalahi, Swardiantara and Ahmad, Tohari and Studiawan, Hudan},
102
+ title = {RoBERTa Model for Drone Flight Log Event Recognition},
103
+ year = {2024},
104
+ publisher = {Hugging Face},
105
+ journal = {Hugging Face Hub}
106
+ }
107
+ ```
108
+ <!--- Describe where people can find more information -->
concept.png ADDED