results_all = pandas.DataFrame({ 'party': votes['class'], 'predicted': pandas.Categorical.from_codes( predicted_party, votes['class'].cat.categories), 'proba(democrat)': party_probabilities[:, 0], 'proba(republican)': party_probabilities[:, 1], }) performance(results_all)