Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- import csv
- import random
- def loadDataset(filename, split, trainingSet=[] , testSet=[]):
- with open(filename, 'rb') as csvfile:
- lines = csv.reader(csvfile)
- dataset = list(lines)
- for x in range(len(dataset)-1):
- for y in range(4):
- dataset[x][y] = float(dataset[x][y])
- if random.random() < split:
- trainingSet.append(dataset[x])
- else:
- testSet.append(dataset[x])
Add Comment
Please, Sign In to add comment