d6376d9c4aa3aea31dcf07e65e35125d7f091c2f,snorkel/learning/disc_learning.py,TFNoiseAwareModel,load,#TFNoiseAwareModel#,83

Before Change

        self._build(**kwargs)
        saver = tf.train.Saver()
        saver.restore(self.session, "{0}.meta".format(model_name))
        if kwargs.get("verbose", False):
            print("[{0}] Loaded model <{1}>".format(self.name, model_name))

After Change

        load_dict = self._build()
        saver = tf.train.Saver(load_dict)
        ckpt = tf.train.get_checkpoint_state("./")
        if ckpt and ckpt.model_checkpoint_path:
            saver.restore(self.session, ckpt.model_checkpoint_path)
            if verbose:
                print("[{0}] Loaded model <{1}>".format(self.name, model_name))
        else:
            raise Exception("[{0}] No model found at <{1}>".format(
                self.name, model_name
            ))


def get_train_idxs(marginals, rebalance=False, split_lo=0.5, split_hi=0.5):
    pos = np.where(marginals < (split_lo - 1e-6))[0]
    neg = np.where(marginals > (split_hi + 1e-6))[0]
    if rebalance:

Instances