generate_data.py

#!/usr/bin/env python
#coding: utf-8

import numpy as np

def generateData(n):
    """
    generates a 2D linearly separable dataset with n samples.
    The thired element of the sample is the label
    """
    xb = (np.random.rand(n) * 2 -1) / 2 - 0.5
    yb = (np.random.rand(n) * 2 -1) / 2 + 0.5
    xr = (np.random.rand(n) * 2 -1) / 2 + 0.5
    yr = (np.random.rand(n) * 2 -1) / 2 - 0.5
    inputs = []
    inputs.extend([[xb[i], yb[i], 1] for i in xrange(n)])
    inputs.extend([[xr[i], yr[i], -1] for i in xrange(n)])
    return inputs