sklearn_ex1.py 文件源码

def case2():
    from sklearn.datasets import make_classification

    x,y = make_classification(n_samples=1000, n_features=2,n_redundant=0,n_informative=1,n_clusters_per_class=1)

    print len(x)
    print len(y)
    print x
    print y
    for i in range(len(x)):
        print x[i],y[i]

    x_data_train = x[:800,:]

    x_data_test = x[800:,:]
    y_data_train = y[:800]
    y_data_test = y[800:]

    print '*'*20
    print x_data_train
    print x_data_test
    print y_data_train
    print y_data_test

    print x[0,0]