Skip to content

Commit 63137b0

Browse files
committed
Fix in TrainerPerfectAgreement to unify the interface with TrainerAgreement.
1 parent 29ceab7 commit 63137b0

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

neural_structured_learning/research/gam/trainer/trainer_agreement.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -920,6 +920,7 @@ class TrainerPerfectAgreement(object):
920920
def __init__(self, data):
921921
self.data = data
922922
self.model = None
923+
self.vars_to_save = []
923924

924925
# Save the true labels in a TensorFlow variable, which is used in the
925926
# create_agreement_prediction function.

0 commit comments

Comments
 (0)