gistfile1.txt

from sklearn import svm

X = mega_results[['D1','D2','D3','D4','D5','D6']]
y = mega_results['Sena']

clf = svm.SVC()
clf.fit(X, y)

import random

tries = 0

while 1:
    tries += 1
    numeros = random.sample(range(1,61), 6)
    if clf.predict(sorted(numeros))[0]:
        print numeros
        break

print tries