Skip to content

Commit f08787d

Browse files
Add RCNN output and logs
1 parent b7071c2 commit f08787d

File tree

3 files changed

+10015
-1
lines changed

3 files changed

+10015
-1
lines changed

code/algorithm/main.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -345,7 +345,7 @@ def parse_args(args):
345345
)
346346
model_params_group.add_argument(
347347
'--epochs',
348-
default=5,
348+
default=9,
349349
type=int,
350350
required=False,
351351
help='The number of epochs to train the model for.'

code/logs/rcnn_linear_bce.txt

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,13 @@
1+
micro_f1: 0.754989, macro_f1: 0.316767, weighted_f1: 0.879010
2+
micro_f1: 0.853110, macro_f1: 0.602525, weighted_f1: 0.879771
3+
micro_f1: 0.888203, macro_f1: 0.697833, weighted_f1: 0.897955
4+
micro_f1: 0.894444, macro_f1: 0.721883, weighted_f1: 0.901083
5+
micro_f1: 0.896963, macro_f1: 0.736876, weighted_f1: 0.902070
6+
micro_f1: 0.897866, macro_f1: 0.739836, weighted_f1: 0.902855
7+
micro_f1: 0.898391, macro_f1: 0.744720, weighted_f1: 0.902566
8+
micro_f1: 0.898354, macro_f1: 0.748265, weighted_f1: 0.901720
9+
micro_f1: 0.898632, macro_f1: 0.749231, weighted_f1: 0.902501
10+
micro_f1: 0.898051, macro_f1: 0.747897, weighted_f1: 0.901669
11+
micro_f1: 0.898201, macro_f1: 0.748179, weighted_f1: 0.901933
12+
micro_f1: 0.897762, macro_f1: 0.746856, weighted_f1: 0.901329
13+
micro_f1: 0.898095, macro_f1: 0.747834, weighted_f1: 0.901711

0 commit comments

Comments
 (0)