From 107e5b1ee728218f5f75694d52e2887074f89639 Mon Sep 17 00:00:00 2001 From: josh herzberg Date: Tue, 17 Jul 2018 20:02:32 -0500 Subject: [PATCH] no more panels --- lib/tagnews/crimetype/tag.py | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) diff --git a/lib/tagnews/crimetype/tag.py b/lib/tagnews/crimetype/tag.py index fae5afb..3f1bb16 100644 --- a/lib/tagnews/crimetype/tag.py +++ b/lib/tagnews/crimetype/tag.py @@ -173,12 +173,12 @@ def get_contributions(self, text): and -0.316201 . -0.853208 """ - p = {} vec = self.vectorizer.transform([text]) - vec_inv = self.vectorizer.inverse_transform(vec) - for i, tag in enumerate(TAGS): - p[tag] = pd.DataFrame( - index=vec_inv, - data={'weight': self.clf.coef_[i, vec.nonzero()[1]]} - ) - return pd.Panel(p) + vec_inv = self.vectorizer.inverse_transform(vec)[1] + + df = pd.DataFrame(columns=vec_inv, index=TAGS) + + for j, word in enumerate(vec_inv): + for i, tag in enumerate(TAGS): + df.loc[tag][word] = mod.clf.coef_[i, vec.nonzero()[1][j]] + return df