Skip to content

Commit 3265019

Browse files
committed
Changing model name.
1 parent 2b68d08 commit 3265019

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

neural_structured_learning/research/gam/experiments/run_train_mnist.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -429,17 +429,21 @@ def main(argv):
429429
model_name += ('_' + FLAGS.hidden_agr) if FLAGS.model_agr == 'mlp' else ''
430430
model_name += '-aggr_' + FLAGS.aggregation_agr_inputs
431431
model_name += ('_' + FLAGS.hidden_aggreg) if FLAGS.hidden_aggreg else ''
432-
model_name += ('-add_%d-conf_%.2f-iter_cls_%d-iter_agr_%d-batch_cls_%d' %
432+
model_name += ('-add_%d-conf_%.2f-iterCls_%d-iterAgr_%d-batchCls_%d' %
433433
(FLAGS.num_samples_to_label, FLAGS.min_confidence_new_label,
434434
FLAGS.max_num_iter_cls, FLAGS.max_num_iter_agr,
435435
FLAGS.batch_size_cls))
436+
model_name += (('-wdecayCls_%.4f' % FLAGS.weight_decay_cls)
437+
if FLAGS.weight_decay_cls else '')
438+
model_name += (('-wdecayAgr_%.4f' % FLAGS.weight_decay_agr)
439+
if FLAGS.weight_decay_agr else '')
436440
model_name += '-LL_%s_LU_%s_UU_%s' % (str(
437441
FLAGS.reg_weight_ll), str(FLAGS.reg_weight_lu), str(FLAGS.reg_weight_uu))
438442
model_name += '-perfAgr' if FLAGS.use_perfect_agreement else ''
439443
model_name += '-perfCls' if FLAGS.use_perfect_classifier else ''
440444
model_name += '-keepProp' if FLAGS.keep_label_proportions else ''
441445
model_name += '-PenNegAgr' if FLAGS.penalize_neg_agr else ''
442-
model_name += '-transduct' if not FLAGS.inductive else ''
446+
model_name += '-transd' if not FLAGS.inductive else ''
443447
model_name += '-L2' if FLAGS.use_l2_cls else '-CE'
444448
model_name += '-seed_' + str(FLAGS.seed)
445449
model_name += FLAGS.experiment_suffix

0 commit comments

Comments
 (0)