Mercurial > repos > bgruening > sklearn_clf_metrics
annotate ml_visualization_ex.py @ 42:9fdff01a34b0 draft
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
| author | bgruening | 
|---|---|
| date | Wed, 09 Aug 2023 12:37:34 +0000 | 
| parents | ed7d6d7c6e93 | 
| children | 21d05f8abca6 | 
| rev | line source | 
|---|---|
| 28 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 1 import argparse | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 2 import json | 
| 35 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 3 import os | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 4 import warnings | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 5 | 
| 32 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 6 import matplotlib | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 7 import matplotlib.pyplot as plt | 
| 28 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 8 import numpy as np | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 9 import pandas as pd | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 10 import plotly | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 11 import plotly.graph_objs as go | 
| 41 
ed7d6d7c6e93
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
36diff
changeset | 12 from galaxy_ml.model_persist import load_model_from_h5 | 
| 
ed7d6d7c6e93
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
36diff
changeset | 13 from galaxy_ml.utils import read_columns, SafeEval | 
| 
ed7d6d7c6e93
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
36diff
changeset | 14 from sklearn.feature_selection._base import SelectorMixin | 
| 
ed7d6d7c6e93
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
36diff
changeset | 15 from sklearn.metrics import ( | 
| 
ed7d6d7c6e93
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
36diff
changeset | 16 auc, | 
| 
ed7d6d7c6e93
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
36diff
changeset | 17 average_precision_score, | 
| 
ed7d6d7c6e93
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
36diff
changeset | 18 precision_recall_curve, | 
| 
ed7d6d7c6e93
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
36diff
changeset | 19 roc_curve, | 
| 
ed7d6d7c6e93
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
36diff
changeset | 20 ) | 
| 28 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 21 from sklearn.pipeline import Pipeline | 
| 41 
ed7d6d7c6e93
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
36diff
changeset | 22 from tensorflow.keras.models import model_from_json | 
| 
ed7d6d7c6e93
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
36diff
changeset | 23 from tensorflow.keras.utils import plot_model | 
| 28 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 24 | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 25 safe_eval = SafeEval() | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 26 | 
| 32 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 27 # plotly default colors | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 28 default_colors = [ | 
| 35 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 29 "#1f77b4", # muted blue | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 30 "#ff7f0e", # safety orange | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 31 "#2ca02c", # cooked asparagus green | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 32 "#d62728", # brick red | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 33 "#9467bd", # muted purple | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 34 "#8c564b", # chestnut brown | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 35 "#e377c2", # raspberry yogurt pink | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 36 "#7f7f7f", # middle gray | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 37 "#bcbd22", # curry yellow-green | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 38 "#17becf", # blue-teal | 
| 32 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 39 ] | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 40 | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 41 | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 42 def visualize_pr_curve_plotly(df1, df2, pos_label, title=None): | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 43 """output pr-curve in html using plotly | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 44 | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 45 df1 : pandas.DataFrame | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 46 Containing y_true | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 47 df2 : pandas.DataFrame | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 48 Containing y_score | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 49 pos_label : None | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 50 The label of positive class | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 51 title : str | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 52 Plot title | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 53 """ | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 54 data = [] | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 55 for idx in range(df1.shape[1]): | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 56 y_true = df1.iloc[:, idx].values | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 57 y_score = df2.iloc[:, idx].values | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 58 | 
| 36 
fc0801adfba1
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ca87db9c038a6fcf96aa39da50f384865fd932ff"
 bgruening parents: 
35diff
changeset | 59 precision, recall, _ = precision_recall_curve( | 
| 
fc0801adfba1
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ca87db9c038a6fcf96aa39da50f384865fd932ff"
 bgruening parents: 
35diff
changeset | 60 y_true, y_score, pos_label=pos_label | 
| 
fc0801adfba1
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ca87db9c038a6fcf96aa39da50f384865fd932ff"
 bgruening parents: 
35diff
changeset | 61 ) | 
| 35 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 62 ap = average_precision_score(y_true, y_score, pos_label=pos_label or 1) | 
| 32 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 63 | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 64 trace = go.Scatter( | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 65 x=recall, | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 66 y=precision, | 
| 35 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 67 mode="lines", | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 68 marker=dict(color=default_colors[idx % len(default_colors)]), | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 69 name="%s (area = %.3f)" % (idx, ap), | 
| 32 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 70 ) | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 71 data.append(trace) | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 72 | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 73 layout = go.Layout( | 
| 35 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 74 xaxis=dict(title="Recall", linecolor="lightslategray", linewidth=1), | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 75 yaxis=dict(title="Precision", linecolor="lightslategray", linewidth=1), | 
| 32 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 76 title=dict( | 
| 35 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 77 text=title or "Precision-Recall Curve", | 
| 32 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 78 x=0.5, | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 79 y=0.92, | 
| 35 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 80 xanchor="center", | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 81 yanchor="top", | 
| 32 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 82 ), | 
| 35 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 83 font=dict(family="sans-serif", size=11), | 
| 32 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 84 # control backgroud colors | 
| 35 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 85 plot_bgcolor="rgba(255,255,255,0)", | 
| 32 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 86 ) | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 87 """ | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 88 legend=dict( | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 89 x=0.95, | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 90 y=0, | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 91 traceorder="normal", | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 92 font=dict( | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 93 family="sans-serif", | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 94 size=9, | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 95 color="black" | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 96 ), | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 97 bgcolor="LightSteelBlue", | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 98 bordercolor="Black", | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 99 borderwidth=2 | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 100 ),""" | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 101 | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 102 fig = go.Figure(data=data, layout=layout) | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 103 | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 104 plotly.offline.plot(fig, filename="output.html", auto_open=False) | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 105 # to be discovered by `from_work_dir` | 
| 35 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 106 os.rename("output.html", "output") | 
| 32 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 107 | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 108 | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 109 def visualize_pr_curve_matplotlib(df1, df2, pos_label, title=None): | 
| 35 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 110 """visualize pr-curve using matplotlib and output svg image""" | 
| 32 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 111 backend = matplotlib.get_backend() | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 112 if "inline" not in backend: | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 113 matplotlib.use("SVG") | 
| 35 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 114 plt.style.use("seaborn-colorblind") | 
| 32 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 115 plt.figure() | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 116 | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 117 for idx in range(df1.shape[1]): | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 118 y_true = df1.iloc[:, idx].values | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 119 y_score = df2.iloc[:, idx].values | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 120 | 
| 36 
fc0801adfba1
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ca87db9c038a6fcf96aa39da50f384865fd932ff"
 bgruening parents: 
35diff
changeset | 121 precision, recall, _ = precision_recall_curve( | 
| 
fc0801adfba1
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ca87db9c038a6fcf96aa39da50f384865fd932ff"
 bgruening parents: 
35diff
changeset | 122 y_true, y_score, pos_label=pos_label | 
| 
fc0801adfba1
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ca87db9c038a6fcf96aa39da50f384865fd932ff"
 bgruening parents: 
35diff
changeset | 123 ) | 
| 35 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 124 ap = average_precision_score(y_true, y_score, pos_label=pos_label or 1) | 
| 32 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 125 | 
| 35 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 126 plt.step( | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 127 recall, | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 128 precision, | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 129 "r-", | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 130 color="black", | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 131 alpha=0.3, | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 132 lw=1, | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 133 where="post", | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 134 label="%s (area = %.3f)" % (idx, ap), | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 135 ) | 
| 32 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 136 | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 137 plt.xlim([0.0, 1.0]) | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 138 plt.ylim([0.0, 1.05]) | 
| 35 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 139 plt.xlabel("Recall") | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 140 plt.ylabel("Precision") | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 141 title = title or "Precision-Recall Curve" | 
| 32 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 142 plt.title(title) | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 143 folder = os.getcwd() | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 144 plt.savefig(os.path.join(folder, "output.svg"), format="svg") | 
| 35 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 145 os.rename(os.path.join(folder, "output.svg"), os.path.join(folder, "output")) | 
| 32 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 146 | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 147 | 
| 35 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 148 def visualize_roc_curve_plotly(df1, df2, pos_label, drop_intermediate=True, title=None): | 
| 32 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 149 """output roc-curve in html using plotly | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 150 | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 151 df1 : pandas.DataFrame | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 152 Containing y_true | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 153 df2 : pandas.DataFrame | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 154 Containing y_score | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 155 pos_label : None | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 156 The label of positive class | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 157 drop_intermediate : bool | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 158 Whether to drop some suboptimal thresholds | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 159 title : str | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 160 Plot title | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 161 """ | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 162 data = [] | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 163 for idx in range(df1.shape[1]): | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 164 y_true = df1.iloc[:, idx].values | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 165 y_score = df2.iloc[:, idx].values | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 166 | 
| 36 
fc0801adfba1
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ca87db9c038a6fcf96aa39da50f384865fd932ff"
 bgruening parents: 
35diff
changeset | 167 fpr, tpr, _ = roc_curve( | 
| 
fc0801adfba1
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ca87db9c038a6fcf96aa39da50f384865fd932ff"
 bgruening parents: 
35diff
changeset | 168 y_true, y_score, pos_label=pos_label, drop_intermediate=drop_intermediate | 
| 
fc0801adfba1
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ca87db9c038a6fcf96aa39da50f384865fd932ff"
 bgruening parents: 
35diff
changeset | 169 ) | 
| 32 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 170 roc_auc = auc(fpr, tpr) | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 171 | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 172 trace = go.Scatter( | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 173 x=fpr, | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 174 y=tpr, | 
| 35 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 175 mode="lines", | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 176 marker=dict(color=default_colors[idx % len(default_colors)]), | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 177 name="%s (area = %.3f)" % (idx, roc_auc), | 
| 32 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 178 ) | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 179 data.append(trace) | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 180 | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 181 layout = go.Layout( | 
| 36 
fc0801adfba1
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ca87db9c038a6fcf96aa39da50f384865fd932ff"
 bgruening parents: 
35diff
changeset | 182 xaxis=dict( | 
| 
fc0801adfba1
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ca87db9c038a6fcf96aa39da50f384865fd932ff"
 bgruening parents: 
35diff
changeset | 183 title="False Positive Rate", linecolor="lightslategray", linewidth=1 | 
| 
fc0801adfba1
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ca87db9c038a6fcf96aa39da50f384865fd932ff"
 bgruening parents: 
35diff
changeset | 184 ), | 
| 35 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 185 yaxis=dict(title="True Positive Rate", linecolor="lightslategray", linewidth=1), | 
| 32 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 186 title=dict( | 
| 35 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 187 text=title or "Receiver Operating Characteristic (ROC) Curve", | 
| 32 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 188 x=0.5, | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 189 y=0.92, | 
| 35 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 190 xanchor="center", | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 191 yanchor="top", | 
| 32 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 192 ), | 
| 35 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 193 font=dict(family="sans-serif", size=11), | 
| 32 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 194 # control backgroud colors | 
| 35 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 195 plot_bgcolor="rgba(255,255,255,0)", | 
| 32 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 196 ) | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 197 """ | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 198 # legend=dict( | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 199 # x=0.95, | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 200 # y=0, | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 201 # traceorder="normal", | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 202 # font=dict( | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 203 # family="sans-serif", | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 204 # size=9, | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 205 # color="black" | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 206 # ), | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 207 # bgcolor="LightSteelBlue", | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 208 # bordercolor="Black", | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 209 # borderwidth=2 | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 210 # ), | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 211 """ | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 212 | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 213 fig = go.Figure(data=data, layout=layout) | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 214 | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 215 plotly.offline.plot(fig, filename="output.html", auto_open=False) | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 216 # to be discovered by `from_work_dir` | 
| 35 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 217 os.rename("output.html", "output") | 
| 32 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 218 | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 219 | 
| 36 
fc0801adfba1
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ca87db9c038a6fcf96aa39da50f384865fd932ff"
 bgruening parents: 
35diff
changeset | 220 def visualize_roc_curve_matplotlib( | 
| 
fc0801adfba1
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ca87db9c038a6fcf96aa39da50f384865fd932ff"
 bgruening parents: 
35diff
changeset | 221 df1, df2, pos_label, drop_intermediate=True, title=None | 
| 
fc0801adfba1
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ca87db9c038a6fcf96aa39da50f384865fd932ff"
 bgruening parents: 
35diff
changeset | 222 ): | 
| 35 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 223 """visualize roc-curve using matplotlib and output svg image""" | 
| 32 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 224 backend = matplotlib.get_backend() | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 225 if "inline" not in backend: | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 226 matplotlib.use("SVG") | 
| 35 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 227 plt.style.use("seaborn-colorblind") | 
| 32 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 228 plt.figure() | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 229 | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 230 for idx in range(df1.shape[1]): | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 231 y_true = df1.iloc[:, idx].values | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 232 y_score = df2.iloc[:, idx].values | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 233 | 
| 36 
fc0801adfba1
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ca87db9c038a6fcf96aa39da50f384865fd932ff"
 bgruening parents: 
35diff
changeset | 234 fpr, tpr, _ = roc_curve( | 
| 
fc0801adfba1
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ca87db9c038a6fcf96aa39da50f384865fd932ff"
 bgruening parents: 
35diff
changeset | 235 y_true, y_score, pos_label=pos_label, drop_intermediate=drop_intermediate | 
| 
fc0801adfba1
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ca87db9c038a6fcf96aa39da50f384865fd932ff"
 bgruening parents: 
35diff
changeset | 236 ) | 
| 32 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 237 roc_auc = auc(fpr, tpr) | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 238 | 
| 35 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 239 plt.step( | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 240 fpr, | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 241 tpr, | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 242 "r-", | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 243 color="black", | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 244 alpha=0.3, | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 245 lw=1, | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 246 where="post", | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 247 label="%s (area = %.3f)" % (idx, roc_auc), | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 248 ) | 
| 32 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 249 | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 250 plt.xlim([0.0, 1.0]) | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 251 plt.ylim([0.0, 1.05]) | 
| 35 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 252 plt.xlabel("False Positive Rate") | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 253 plt.ylabel("True Positive Rate") | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 254 title = title or "Receiver Operating Characteristic (ROC) Curve" | 
| 32 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 255 plt.title(title) | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 256 folder = os.getcwd() | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 257 plt.savefig(os.path.join(folder, "output.svg"), format="svg") | 
| 35 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 258 os.rename(os.path.join(folder, "output.svg"), os.path.join(folder, "output")) | 
| 32 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 259 | 
| 28 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 260 | 
| 35 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 261 def main( | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 262 inputs, | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 263 infile_estimator=None, | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 264 infile1=None, | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 265 infile2=None, | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 266 outfile_result=None, | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 267 outfile_object=None, | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 268 groups=None, | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 269 ref_seq=None, | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 270 intervals=None, | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 271 targets=None, | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 272 fasta_path=None, | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 273 model_config=None, | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 274 ): | 
| 28 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 275 """ | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 276 Parameter | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 277 --------- | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 278 inputs : str | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 279 File path to galaxy tool parameter | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 280 | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 281 infile_estimator : str, default is None | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 282 File path to estimator | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 283 | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 284 infile1 : str, default is None | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 285 File path to dataset containing features or true labels. | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 286 | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 287 infile2 : str, default is None | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 288 File path to dataset containing target values or predicted | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 289 probabilities. | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 290 | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 291 outfile_result : str, default is None | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 292 File path to save the results, either cv_results or test result | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 293 | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 294 outfile_object : str, default is None | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 295 File path to save searchCV object | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 296 | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 297 groups : str, default is None | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 298 File path to dataset containing groups labels | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 299 | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 300 ref_seq : str, default is None | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 301 File path to dataset containing genome sequence file | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 302 | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 303 intervals : str, default is None | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 304 File path to dataset containing interval file | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 305 | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 306 targets : str, default is None | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 307 File path to dataset compressed target bed file | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 308 | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 309 fasta_path : str, default is None | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 310 File path to dataset containing fasta file | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 311 | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 312 model_config : str, default is None | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 313 File path to dataset containing JSON config for neural networks | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 314 """ | 
| 35 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 315 warnings.simplefilter("ignore") | 
| 28 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 316 | 
| 35 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 317 with open(inputs, "r") as param_handler: | 
| 28 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 318 params = json.load(param_handler) | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 319 | 
| 35 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 320 title = params["plotting_selection"]["title"].strip() | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 321 plot_type = params["plotting_selection"]["plot_type"] | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 322 plot_format = params["plotting_selection"]["plot_format"] | 
| 32 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 323 | 
| 35 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 324 if plot_type == "feature_importances": | 
| 41 
ed7d6d7c6e93
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
36diff
changeset | 325 estimator = load_model_from_h5(infile_estimator) | 
| 28 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 326 | 
| 36 
fc0801adfba1
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ca87db9c038a6fcf96aa39da50f384865fd932ff"
 bgruening parents: 
35diff
changeset | 327 column_option = params["plotting_selection"]["column_selector_options"][ | 
| 
fc0801adfba1
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ca87db9c038a6fcf96aa39da50f384865fd932ff"
 bgruening parents: 
35diff
changeset | 328 "selected_column_selector_option" | 
| 
fc0801adfba1
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ca87db9c038a6fcf96aa39da50f384865fd932ff"
 bgruening parents: 
35diff
changeset | 329 ] | 
| 35 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 330 if column_option in [ | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 331 "by_index_number", | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 332 "all_but_by_index_number", | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 333 "by_header_name", | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 334 "all_but_by_header_name", | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 335 ]: | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 336 c = params["plotting_selection"]["column_selector_options"]["col1"] | 
| 28 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 337 else: | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 338 c = None | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 339 | 
| 35 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 340 _, input_df = read_columns( | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 341 infile1, | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 342 c=c, | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 343 c_option=column_option, | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 344 return_df=True, | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 345 sep="\t", | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 346 header="infer", | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 347 parse_dates=True, | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 348 ) | 
| 28 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 349 | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 350 feature_names = input_df.columns.values | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 351 | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 352 if isinstance(estimator, Pipeline): | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 353 for st in estimator.steps[:-1]: | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 354 if isinstance(st[-1], SelectorMixin): | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 355 mask = st[-1].get_support() | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 356 feature_names = feature_names[mask] | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 357 estimator = estimator.steps[-1][-1] | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 358 | 
| 35 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 359 if hasattr(estimator, "coef_"): | 
| 28 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 360 coefs = estimator.coef_ | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 361 else: | 
| 35 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 362 coefs = getattr(estimator, "feature_importances_", None) | 
| 28 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 363 if coefs is None: | 
| 36 
fc0801adfba1
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ca87db9c038a6fcf96aa39da50f384865fd932ff"
 bgruening parents: 
35diff
changeset | 364 raise RuntimeError( | 
| 
fc0801adfba1
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ca87db9c038a6fcf96aa39da50f384865fd932ff"
 bgruening parents: 
35diff
changeset | 365 "The classifier does not expose " | 
| 
fc0801adfba1
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ca87db9c038a6fcf96aa39da50f384865fd932ff"
 bgruening parents: 
35diff
changeset | 366 '"coef_" or "feature_importances_" ' | 
| 
fc0801adfba1
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ca87db9c038a6fcf96aa39da50f384865fd932ff"
 bgruening parents: 
35diff
changeset | 367 "attributes" | 
| 
fc0801adfba1
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ca87db9c038a6fcf96aa39da50f384865fd932ff"
 bgruening parents: 
35diff
changeset | 368 ) | 
| 28 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 369 | 
| 35 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 370 threshold = params["plotting_selection"]["threshold"] | 
| 28 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 371 if threshold is not None: | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 372 mask = (coefs > threshold) | (coefs < -threshold) | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 373 coefs = coefs[mask] | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 374 feature_names = feature_names[mask] | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 375 | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 376 # sort | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 377 indices = np.argsort(coefs)[::-1] | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 378 | 
| 35 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 379 trace = go.Bar(x=feature_names[indices], y=coefs[indices]) | 
| 28 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 380 layout = go.Layout(title=title or "Feature Importances") | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 381 fig = go.Figure(data=[trace], layout=layout) | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 382 | 
| 35 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 383 plotly.offline.plot(fig, filename="output.html", auto_open=False) | 
| 32 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 384 # to be discovered by `from_work_dir` | 
| 35 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 385 os.rename("output.html", "output") | 
| 32 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 386 | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 387 return 0 | 
| 28 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 388 | 
| 35 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 389 elif plot_type in ("pr_curve", "roc_curve"): | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 390 df1 = pd.read_csv(infile1, sep="\t", header="infer") | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 391 df2 = pd.read_csv(infile2, sep="\t", header="infer").astype(np.float32) | 
| 32 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 392 | 
| 35 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 393 minimum = params["plotting_selection"]["report_minimum_n_positives"] | 
| 32 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 394 # filter out columns whose n_positives is beblow the threhold | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 395 if minimum: | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 396 mask = df1.sum(axis=0) >= minimum | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 397 df1 = df1.loc[:, mask] | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 398 df2 = df2.loc[:, mask] | 
| 28 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 399 | 
| 35 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 400 pos_label = params["plotting_selection"]["pos_label"].strip() or None | 
| 28 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 401 | 
| 35 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 402 if plot_type == "pr_curve": | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 403 if plot_format == "plotly_html": | 
| 32 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 404 visualize_pr_curve_plotly(df1, df2, pos_label, title=title) | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 405 else: | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 406 visualize_pr_curve_matplotlib(df1, df2, pos_label, title) | 
| 35 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 407 else: # 'roc_curve' | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 408 drop_intermediate = params["plotting_selection"]["drop_intermediate"] | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 409 if plot_format == "plotly_html": | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 410 visualize_roc_curve_plotly( | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 411 df1, | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 412 df2, | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 413 pos_label, | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 414 drop_intermediate=drop_intermediate, | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 415 title=title, | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 416 ) | 
| 32 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 417 else: | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 418 visualize_roc_curve_matplotlib( | 
| 35 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 419 df1, | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 420 df2, | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 421 pos_label, | 
| 32 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 422 drop_intermediate=drop_intermediate, | 
| 35 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 423 title=title, | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 424 ) | 
| 28 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 425 | 
| 32 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 426 return 0 | 
| 28 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 427 | 
| 35 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 428 elif plot_type == "rfecv_gridscores": | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 429 input_df = pd.read_csv(infile1, sep="\t", header="infer") | 
| 28 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 430 scores = input_df.iloc[:, 0] | 
| 35 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 431 steps = params["plotting_selection"]["steps"].strip() | 
| 28 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 432 steps = safe_eval(steps) | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 433 | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 434 data = go.Scatter( | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 435 x=list(range(len(scores))), | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 436 y=scores, | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 437 text=[str(_) for _ in steps] if steps else None, | 
| 35 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 438 mode="lines", | 
| 28 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 439 ) | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 440 layout = go.Layout( | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 441 xaxis=dict(title="Number of features selected"), | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 442 yaxis=dict(title="Cross validation score"), | 
| 36 
fc0801adfba1
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ca87db9c038a6fcf96aa39da50f384865fd932ff"
 bgruening parents: 
35diff
changeset | 443 title=dict( | 
| 
fc0801adfba1
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ca87db9c038a6fcf96aa39da50f384865fd932ff"
 bgruening parents: 
35diff
changeset | 444 text=title or None, x=0.5, y=0.92, xanchor="center", yanchor="top" | 
| 
fc0801adfba1
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ca87db9c038a6fcf96aa39da50f384865fd932ff"
 bgruening parents: 
35diff
changeset | 445 ), | 
| 35 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 446 font=dict(family="sans-serif", size=11), | 
| 32 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 447 # control backgroud colors | 
| 35 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 448 plot_bgcolor="rgba(255,255,255,0)", | 
| 28 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 449 ) | 
| 32 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 450 """ | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 451 # legend=dict( | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 452 # x=0.95, | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 453 # y=0, | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 454 # traceorder="normal", | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 455 # font=dict( | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 456 # family="sans-serif", | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 457 # size=9, | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 458 # color="black" | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 459 # ), | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 460 # bgcolor="LightSteelBlue", | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 461 # bordercolor="Black", | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 462 # borderwidth=2 | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 463 # ), | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 464 """ | 
| 28 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 465 | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 466 fig = go.Figure(data=[data], layout=layout) | 
| 35 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 467 plotly.offline.plot(fig, filename="output.html", auto_open=False) | 
| 32 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 468 # to be discovered by `from_work_dir` | 
| 35 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 469 os.rename("output.html", "output") | 
| 32 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 470 | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 471 return 0 | 
| 28 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 472 | 
| 35 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 473 elif plot_type == "learning_curve": | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 474 input_df = pd.read_csv(infile1, sep="\t", header="infer") | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 475 plot_std_err = params["plotting_selection"]["plot_std_err"] | 
| 28 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 476 data1 = go.Scatter( | 
| 35 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 477 x=input_df["train_sizes_abs"], | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 478 y=input_df["mean_train_scores"], | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 479 error_y=dict(array=input_df["std_train_scores"]) if plot_std_err else None, | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 480 mode="lines", | 
| 28 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 481 name="Train Scores", | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 482 ) | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 483 data2 = go.Scatter( | 
| 35 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 484 x=input_df["train_sizes_abs"], | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 485 y=input_df["mean_test_scores"], | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 486 error_y=dict(array=input_df["std_test_scores"]) if plot_std_err else None, | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 487 mode="lines", | 
| 28 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 488 name="Test Scores", | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 489 ) | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 490 layout = dict( | 
| 35 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 491 xaxis=dict(title="No. of samples"), | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 492 yaxis=dict(title="Performance Score"), | 
| 32 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 493 # modify these configurations to customize image | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 494 title=dict( | 
| 35 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 495 text=title or "Learning Curve", | 
| 32 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 496 x=0.5, | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 497 y=0.92, | 
| 35 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 498 xanchor="center", | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 499 yanchor="top", | 
| 32 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 500 ), | 
| 35 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 501 font=dict(family="sans-serif", size=11), | 
| 32 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 502 # control backgroud colors | 
| 35 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 503 plot_bgcolor="rgba(255,255,255,0)", | 
| 28 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 504 ) | 
| 32 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 505 """ | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 506 # legend=dict( | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 507 # x=0.95, | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 508 # y=0, | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 509 # traceorder="normal", | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 510 # font=dict( | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 511 # family="sans-serif", | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 512 # size=9, | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 513 # color="black" | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 514 # ), | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 515 # bgcolor="LightSteelBlue", | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 516 # bordercolor="Black", | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 517 # borderwidth=2 | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 518 # ), | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 519 """ | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 520 | 
| 28 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 521 fig = go.Figure(data=[data1, data2], layout=layout) | 
| 35 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 522 plotly.offline.plot(fig, filename="output.html", auto_open=False) | 
| 32 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 523 # to be discovered by `from_work_dir` | 
| 35 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 524 os.rename("output.html", "output") | 
| 32 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 525 | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 526 return 0 | 
| 28 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 527 | 
| 35 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 528 elif plot_type == "keras_plot_model": | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 529 with open(model_config, "r") as f: | 
| 28 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 530 model_str = f.read() | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 531 model = model_from_json(model_str) | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 532 plot_model(model, to_file="output.png") | 
| 35 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 533 os.rename("output.png", "output") | 
| 28 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 534 | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 535 return 0 | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 536 | 
| 32 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 537 # save pdf file to disk | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 538 # fig.write_image("image.pdf", format='pdf') | 
| 
e801d2034575
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: 
29diff
changeset | 539 # fig.write_image("image.pdf", format='pdf', width=340*2, height=226*2) | 
| 28 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 540 | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 541 | 
| 35 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 542 if __name__ == "__main__": | 
| 28 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 543 aparser = argparse.ArgumentParser() | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 544 aparser.add_argument("-i", "--inputs", dest="inputs", required=True) | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 545 aparser.add_argument("-e", "--estimator", dest="infile_estimator") | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 546 aparser.add_argument("-X", "--infile1", dest="infile1") | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 547 aparser.add_argument("-y", "--infile2", dest="infile2") | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 548 aparser.add_argument("-O", "--outfile_result", dest="outfile_result") | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 549 aparser.add_argument("-o", "--outfile_object", dest="outfile_object") | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 550 aparser.add_argument("-g", "--groups", dest="groups") | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 551 aparser.add_argument("-r", "--ref_seq", dest="ref_seq") | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 552 aparser.add_argument("-b", "--intervals", dest="intervals") | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 553 aparser.add_argument("-t", "--targets", dest="targets") | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 554 aparser.add_argument("-f", "--fasta_path", dest="fasta_path") | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 555 aparser.add_argument("-c", "--model_config", dest="model_config") | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 556 args = aparser.parse_args() | 
| 
c077c537cb67
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: diff
changeset | 557 | 
| 35 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 558 main( | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 559 args.inputs, | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 560 args.infile_estimator, | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 561 args.infile1, | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 562 args.infile2, | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 563 args.outfile_result, | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 564 outfile_object=args.outfile_object, | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 565 groups=args.groups, | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 566 ref_seq=args.ref_seq, | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 567 intervals=args.intervals, | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 568 targets=args.targets, | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 569 fasta_path=args.fasta_path, | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 570 model_config=args.model_config, | 
| 
3c5034b0d775
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
34diff
changeset | 571 ) | 
