Skip to content

Commit 79aba0d

Browse files
committed
Fix in TrainerPerfectClassification to unify the interface with TrainerClassification.
1 parent 63137b0 commit 79aba0d

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

neural_structured_learning/research/gam/trainer/trainer_classification.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -756,6 +756,7 @@ class TrainerPerfectClassification(Trainer):
756756

757757
def __init__(self, data):
758758
self.data = data
759+
self.vars_to_save = []
759760

760761
def train(self, unused_data, unused_session=None, **unused_kwargs):
761762
logging.info('Perfect classifier, no need to train...')

0 commit comments

Comments
 (0)