2ab8c1444facbd46df8767a5badda5b9f1a50c29,tpot/operators/classifiers/base.py,Classifier,_train_and_predict,#Classifier#Any#Any#,37

Before Change

        // Fit classifier to the data set
        clf.fit(self.training_features, self.training_classes)

        all_features = input_df.drop(self.non_feature_columns, axis=1).values
        input_df.loc[:, "guess"] = clf.predict(all_features)

        // Store the guesses as a synthetic feature
        return self._add_synth_feature(input_df, operator_args)

After Change

        clf.fit(self.training_features, self.training_classes)

        all_features = np.copy(input_matrix)
        np.delete(all_features, non_feature_columns, axis=1)
        input_matrix[:, GUESS_COL] = clf.predict(all_features)
        // Store the guesses as a synthetic feature
        input_matrix[:, :-1] = input_matrix[GUESS_COL]

Instances