Ole commited on
Commit
788f1b4
1 Parent(s): bb27442

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +12 -1
README.md CHANGED
@@ -7,4 +7,15 @@ pipeline_tag: text-classification
7
  tags:
8
  - code
9
  license: cc-by-nc-sa-4.0
10
- ---
 
 
 
 
 
 
 
 
 
 
 
 
7
  tags:
8
  - code
9
  license: cc-by-nc-sa-4.0
10
+ datasets:
11
+ - code_search_net
12
+ ---
13
+
14
+ With this model, the return value can be classified based on the method name only.
15
+ Six classes are available for this purpose.
16
+ The model answers with the label indexes and the associated probability.
17
+ The corresponding class name must then be taken from the following list: ['None', 'Number', 'Boolean', 'String', 'Object', 'Collection'].
18
+
19
+ The corresponding huggingface space can be found at https://huggingface.co/spaces/UDE-SE/ReturnTypePredictor.
20
+
21
+ Good luck trying it out!