We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 82dd413 commit 3e73cb8Copy full SHA for 3e73cb8
neural_structured_learning/research/gam/experiments/run_train_gam_graph.py
@@ -323,6 +323,8 @@ def main(argv):
323
model_name += '-perfCls' if FLAGS.use_perfect_classifier else ''
324
model_name += '-keepProp' if FLAGS.keep_label_proportions else ''
325
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 ''
328
model_name += '-transd' if not FLAGS.inductive else ''
329
model_name += '-L2' if FLAGS.use_l2_cls else '-CE'
330
model_name += '-graph' if FLAGS.use_graph else '-noGraph'
0 commit comments