c29e21cefeaa1461474c26720b3a5e293480c797,snorkel/learning/disc_learning.py,TFNoiseAwareModel,train,#TFNoiseAwareModel#Any#Any#Any#Any#Any#Any#Any#Any#Any#Any#Any#Any#,101

Before Change

            diffs = Y_train.max(axis=1) - Y_train.min(axis=1)
            train_idxs = np.where(diffs > 1e-6)[0]
        X_train = [X_train[j] for j in train_idxs] if self.representation \
            else sparse.csr_matrix(X_train[train_idxs, :])
        Y_train = Y_train[train_idxs]

        // Create new graph, build network, and start session
        self._build_new_graph_session(**kwargs)

After Change

        X_train = [X_train[j] for j in train_idxs] if self.representation \
            else X_train[train_idxs, :]
        if isinstance(X_train, csr_AnnotationMatrix):
            X_train = sparse.csr_matrix(X_train[train_idxs, :])
        Y_train = Y_train[train_idxs]

        // Create new graph, build network, and start session
        self._build_new_graph_session(**kwargs)

Instances