18 lines
435 B
Python
18 lines
435 B
Python
import numpy as np
|
|
import pandas as pd
|
|
|
|
|
|
def load_data():
|
|
df = pd.read_csv('data/sample.txt', delimiter=',', header=None).astype(np.float32)
|
|
|
|
#df = pd.read_csv('data/ex1data1.txt', delimiter=',', header=None).astype(np.float32)
|
|
|
|
#df = pd.read_csv('data/train.csv', delimiter=',', comment='#').astype(np.float32)
|
|
#df[0] = df['x']
|
|
#df[1] = df['y']
|
|
|
|
df[2] = pd.Series([1]*len(df[0]))
|
|
df = df.reindex(columns=[1, 2, 0])
|
|
|
|
return df
|