Skip to content

Commit

Permalink
fixed parameter passing to seaborn by first creating a pandas
Browse files Browse the repository at this point in the history
frame. working on issue #5
  • Loading branch information
sergeyplis committed Apr 7, 2015
1 parent 8f1e6a0 commit 69612a3
Showing 1 changed file with 4 additions and 2 deletions.
6 changes: 4 additions & 2 deletions polyssifier.py
Original file line number Diff line number Diff line change
Expand Up @@ -107,7 +107,8 @@ def make_classifiers(data_shape, ksplit) :
n_jobs=PROCESSORS),
"Logistic Regression": LogisticRegression(),
"Naive Bayes": GaussianNB(),
"LDA": LDA()}
"LDA": LDA()
}

params = {
"Nearest Neighbors": [{"n_neighbors": [1, 5, 10, 20]}],
Expand Down Expand Up @@ -384,7 +385,8 @@ def main(source_dir, ksplit, out_dir, data_pattern, label_pattern):

pl.figure(figsize=[10,6])
ax=pl.gca()
sb.barplot(np.array(NAMES), dscore, palette="Paired")
ds = pd.DataFrame(dscore.T, columns=np.array(NAMES))
sb.barplot(data=ds, palette="Paired")
ax.set_xticks(np.arange(len(NAMES)))
ax.set_xticklabels(NAMES, rotation=30)
ax.set_ylabel("classification AUC")
Expand Down

0 comments on commit 69612a3

Please sign in to comment.