mochodek commited on
Commit
938e4d8
1 Parent(s): 59208d1

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +16 -0
README.md CHANGED
@@ -1,3 +1,19 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
  license: apache-2.0
3
  ---
 
1
+ The model is a port of our CommentBERT model from the paper:
2
+
3
+ ```
4
+ @inproceedings{ochodek2022automated,
5
+ title={Automated code review comment classification to improve modern code reviews},
6
+ author={Ochodek, Miroslaw and Staron, Miroslaw and Meding, Wilhelm and S{\"o}der, Ola},
7
+ booktitle={International Conference on Software Quality},
8
+ pages={23--40},
9
+ year={2022},
10
+ organization={Springer}
11
+ }
12
+ ```
13
+
14
+ The original model was implemented in Keras with two outputs - comment-purpose and subject-purpose. Here, we divided it into two separate model with one output each.
15
+
16
+
17
  ---
18
  license: apache-2.0
19
  ---