1 files changed, 157 insertions, 0 deletions
diff --git a/sourcecodes/ts_plotly.py b/sourcecodes/ts_plotly.py
new file mode 100644
index 00000000..bd7d1405
--- /dev/null
+++ b/sourcecodes/ts_plotly.py
@@ -0,0 +1,157 @@
+#!/home/jziebart/python/Python-2.7.15/python
+import os
+import sys
+
+#sys.path.append('/home/jziebart/.local/bin')
+#sys.path.append('/home/jziebart/.local/lib')
+
+import plotly
+import plotly.graph_objs as go
+import csv
+import string
+
+netID = sys.argv[-1]
+outfile = netID+"ts_plotly.html"
+
+ 
+filename=netID+"ts_output.txt"
+f=open(filename,"r")
+#Read the first line to get the variable name
+line=f.readline()
+line = map(string.strip,line.strip().split(" "))
+varName = line[-1]
+
+print varName
+
+#Read type file to determine if it is continuous or discrete
+typefile = netID+"type.txt"
+tf=open(typefile,"r")
+line=tf.readline()
+varnames = map(string.strip,line.strip().split("\t"))
+print varnames
+line=tf.readline()
+vartypes = map(string.strip,line.strip().split("\t"))
+varindex = varnames.index(varName)
+cd_type = int(vartypes[varindex])
+
+
+if cd_type == 1:
+    #Make scatterplot for continuous_data
+    #Read introductory lines from file
+    for i in range(6):
+        line = f.readline()
+    #Read the data
+    x = []
+    y = []
+    line = f.readline()
+    while line:
+        line = map(string.strip,line.strip().split("\t"))
+        if line[1] != 'NA':     
+            x.append(float(line[1]))
+            y.append(float(line[2]))
+        line=f.readline()
+
+    data = [go.Scatter(x=x,y=y,mode='markers')]
+    layout = go.Layout(
+        xaxis=dict(
+            autorange=True,
+            title='Actual values',
+            titlefont=dict(
+                family='Arial, sans-serif',
+                size=18,
+                color='black'
+                ),
+            ),
+        yaxis=dict(
+            autorange=True,
+            title='Predicted values',
+            titlefont=dict(
+                family='Arial, sans-serif',
+                size=18,
+                color='black'
+                ),
+            )
+        )
+    fig = go.Figure(data=data, layout = layout)
+    plotly.offline.plot(fig,filename=outfile)
+
+else:
+    #Make bar chart for discrete data
+    #Read introductory lines from file
+    for i in range(5):
+        line = f.readline()
+    #Get names of states
+    line = map(string.strip,line.strip().split("\t"))
+    states = line[2:]
+    #Read the data
+    actual = []
+    predicted = []
+    line = f.readline()
+    while line:
+        line = map(string.strip,line.strip().split("\t"))
+        if line[1] != 'NA':
+            actual.append(line[1])
+            predict_x = line[2:]
+            predict_x = [float(x) for x in predict_x]
+            max_value = max(predict_x)
+            max_index = predict_x.index(max_value)
+            predicted.append(states[max_index])
+            #check if multiple states are equally likely to be predicted states
+            #I am not going to count these as being predicted here
+            max_items = [x for x in predict_x if (abs(x-max_value) < 0.000001)]
+            if len(max_items) > 1:
+                predicted.pop()
+                actual.pop()
+        line=f.readline()
+    
+    #Go through states and get number of true positives, false positives, and false negatives
+    tp_all = []
+    fp_all = []
+    fn_all = []
+    for state in states:
+        tp = 0
+        fp = 0
+        fn = 0
+        for i in range(len(actual)):
+            actual_i = actual[i]
+            predicted_i = predicted[i]
+            if actual_i == state:
+                if predicted_i == state:
+                    tp = tp + 1
+                else:
+                    fn = fn + 1
+            elif predicted_i == state:
+                fp = fp + 1
+        tp_all.append(tp)
+        fn_all.append(fn)
+        fp_all.append(fp)
+    print tp_all
+    print fn_all
+    print fp_all
+    trace1 = go.Bar(x=states,y=tp_all,name="True Positives")
+    trace2 = go.Bar(x=states,y=fn_all,name="False Negatives")
+    trace3 = go.Bar(x=states,y=fp_all,name="False Positives")
+    data = [trace1,trace2,trace3]
+    layout = go.Layout(
+        barmode='group',
+        xaxis=dict(
+            autorange=True,
+            title='State',
+            titlefont=dict(
+                family='Arial, sans-serif',
+                size=18,
+                color='black'
+                ),
+            ),
+        yaxis=dict(
+            autorange=True,
+            title='Number of cases',
+            titlefont=dict(
+                family='Arial, sans-serif',
+                size=18,
+                color='black'
+                ),
+            )
+    )
+    fig = go.Figure(data=data, layout = layout)
+    plotly.offline.plot(fig,filename=outfile)