In [3]:

    

'''
The following will produce a line graph representing
the time at which each learner's parameters converged
'''


    

    
Out[3]:





"\nThe following will produce a line graph representing\nthe time at which each learner's parameters converged\n"

In [4]:

    

# Standard import for pandas, numpy and matplot
import pandas as pd
import numpy as np
import matplotlib.pyplot as plt


    

In [5]:

    

%matplotlib inline


    

In [6]:

    

# Import and view data from out.csv
df = pd.read_csv('out.csv')
df.head()


    

    
Out[6]:






  

    

      

      
p1
      
p2
      
p3
      
p4
      
p5
      
p6
      
p7
      
p8
      
p9
      
p10
      
p11
      
p12
      
p13
    
  
  

    

      
0
      
1
      
1
      
1
      
1
      
1
      
1
      
1
      
1
      
1
      
1
      
1
      
4
      
1
    
    

      
1
      
1
      
1
      
1
      
1
      
1
      
1
      
1
      
4
      
1
      
1
      
4
      
1
      
1
    
    

      
2
      
1
      
1
      
1
      
1
      
1
      
1
      
1
      
2
      
1
      
1
      
2
      
1
      
1
    
    

      
3
      
1
      
1
      
1
      
1
      
1
      
1
      
1
      
3
      
1
      
1
      
3
      
1
      
1
    
    

      
4
      
1
      
1
      
1
      
3
      
1
      
1
      
1
      
1
      
1
      
1
      
1
      
3
      
1
    
  

In [7]:

    

# Get the number of rows in the dataframe
numLearners = df.shape[0]

# Create a list of lists, each inner list represents
# a parameter and the sentence on which it converged on
rowList = []
for i in range(0, numLearners):
    # Use iloc to grab a row
    # and convert it to a list
    l = list(df.iloc[[i]].values.flatten())
    dbList = []
    for j in range(0, 13):
        dbList.append([l[j], j+1])
    dbList.sort(key=lambda x: x[0])
    rowList.append(dbList)


    

In [14]:

    

# Returns one of five colors based on the current
# numbers's last digit. This will be assigned
# to a line in the graph representing a learner
def get_line_color(num):
    lastDigit = num % 10
    if lastDigit == 0 or lastDigit == 5:
        return "blue"
    elif lastDigit == 1 or lastDigit == 6:
        return "green"
    elif lastDigit == 2 or lastDigit == 7:
        return "red"
    elif lastDigit == 3 or lastDigit == 8:
        return "yellow"
    else: # lastDigit == 4 or lastDigit == 9
        return "purple"


    

In [15]:

    

# Stores every completed line that will appear
# on the graph
data = []
for i in range(0, numLearners):
    # Variables used to store the x-axis
    # and y axis data of each learner
    xData = []
    yData = []
    
    # The appropiate data is added,
    # xData stores the sentence number on which
    # each parameter converged
    # yData stores the corresponding parameter number
    for j in range(0, 13):
        xData.append(rowList[i][j][0])
        yData.append(rowList[i][j][1])
    
    # Every three elements of data will represent one line on the graph
    data.append(xData)
    data.append(yData)
    data.append(get_line_color(i))


    

In [16]:

    

# Create a figure of size 8x6 inches, 80 dots per inch
plt.figure(figsize=(8, 6), dpi=80)

# Add graph title and axis labels
plt.title("Convergence Time of Parameters")
plt.xlabel("Time")
plt.ylabel("Parameters")

for i in range(0, numLearners, 3):
    plt.plot(data[i], data[i+1], color=data[i+2], linewidth=1.0, marker='.', linestyle="-")

# Set x limits
plt.xlim(0, 13)

# Set x ticks
plt.xticks(np.linspace(0, 13, 14, endpoint=True))

# Set y limits
plt.ylim(1, 13)

# Set y ticks
plt.yticks(np.linspace(1, 13, 13, endpoint=True))

plt.show()


    

In [17]:

    

figure = plt.figure()
figure.savefig('pset.pdf')


    

    






<matplotlib.figure.Figure at 0x7f71635d9990>

In [ ]: