results = pandas.DataFrame({ 'party': votes['class'][:10], 'predicted': pandas.Categorical.from_codes( predicted_party, votes['class'][:10].cat.categories), 'proba(Republican)': party_probabilities[:, 0], 'proba(Democrat)': party_probabilities[:, 1], }) results