In [1]:

    

%matplotlib inline

import matplotlib.pyplot as plt


    

In [2]:

    

import csv

x1 = []
x2 = []
pred = []

with open('../tests/data/test.mat', encoding='utf-8') as csvfile:
    reader = csv.reader(csvfile, delimiter=',')
    for row in reader:
        x1.append(float(row[0]))
        x2.append(float(row[1]))
with open('../tests/data/prediction.mat', encoding='utf-8') as csvfile:
    reader = csv.reader(csvfile, delimiter=',')
    for row in reader:
        pred.append(float(row[0]))
        
plt.figure(figsize=(15,6))
plt.title("Predictions: red=1, blue=-1")
plt.scatter(x1, x2, s=100, cmap='Paired', marker='x', c=pred)


    

    
Out[2]:





<matplotlib.collections.PathCollection at 0x7fb8531fad30>