Skip to content

Commit 3e73cb8

Browse files
committed
Add term to model name.
1 parent 82dd413 commit 3e73cb8

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

neural_structured_learning/research/gam/experiments/run_train_gam_graph.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -323,6 +323,8 @@ def main(argv):
323323
model_name += '-perfCls' if FLAGS.use_perfect_classifier else ''
324324
model_name += '-keepProp' if FLAGS.keep_label_proportions else ''
325325
model_name += '-PenNegAgr' if FLAGS.penalize_neg_agr else ''
326+
model_name += '-VAT' if FLAGS.reg_weight_vat > 0 else ''
327+
model_name += 'ENT' if FLAGS.reg_weight_vat > 0 and FLAGS.use_ent_min else ''
326328
model_name += '-transd' if not FLAGS.inductive else ''
327329
model_name += '-L2' if FLAGS.use_l2_cls else '-CE'
328330
model_name += '-graph' if FLAGS.use_graph else '-noGraph'

0 commit comments

Comments
 (0)