se2p
/

b-fein commited on
Commit
cccea6b
·
verified ·
1 Parent(s): 7769662
1/abstract/model-best.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:48fc9466e33d6973f55bb6ed68218f4fdad4b9d846469fea4a836436e4515da6
3
+ size 8852514453
1/raw/model-best.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a1257e18c941d4a24b723dcca410577085b2762a9c840e9c55cca24c6fa08110
3
+ size 8852514453
1/test-method/model-best.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fe7f8237a05fec853acd26a0164747779906200f78fd1b059f402161eaecace0
3
+ size 8852514453
10/abstract/model-best.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ebe96b709f7b38bc70411f39a2cf71c3190f855fe771aff9631a160f6c1a9e74
3
+ size 8852514453
10/raw/model-best.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:888af0dd160f71d435cc953e8dd271f40536f7585daf4f5151f6fe4fd94b6460
3
+ size 8852514453
10/test-method/model-best.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d955ff744dcdb215b82b2b64cc29dc95d23b480e5ed159504d4b1183541a2721
3
+ size 8852514453
5/abstract/model-best.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:deb1ef63b256b59964e3106ec62a1f9dcd3fcc8e5b4c988435ad9bb0443abb89
3
+ size 8852514453
5/raw/model-best.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1a59da0ba5d40f33324ae55dd9d8b5b6806b66bbcab7e82a07b9483cf5f90cf5
3
+ size 8852514453
5/test-method/model-best.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d668e5c1d007abaff279799a7fdb355d3dcef9ecc366a6766a6b3fbcf87ca6e0
3
+ size 8852514453
README.md CHANGED
@@ -1,3 +1,22 @@
1
  ---
2
  license: cc-by-4.0
 
 
3
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
  license: cc-by-4.0
3
+ base_model:
4
+ - Salesforce/codet5-large
5
  ---
6
+
7
+ # AsserT5: Test Assertion Generation Using a Fine-Tuned Code Language Model
8
+
9
+ Part of the replication package for our paper at AST 2025 (ToDo: add doi-link when known).
10
+ To be used in combination with our main replication package at https://doi.org/10.5281/zenodo.14703162.
11
+
12
+ AsserT5 is a fine-tuned [CodeT5](https://huggingface.co/Salesfoce/codet5-large) trained to generate assertion statements for Java JUnit test cases.
13
+ It was trained on an extended variant of the [methods2test](https://github.com/microsoft/methods2test) dataset.
14
+
15
+
16
+ ## Structure
17
+
18
+ - The top-level number indicates the maximum number of assertions allowed per test case in the training dataet.
19
+ - The next level below indicates the model variant:
20
+ - `abstract`: Identifiers in the data are replaced with abstract tokens.
21
+ - `raw`: The source code is tokenised as-is.
22
+ - `test-method`: The model is trained only on the test case code rather than test case + focal method pairs.