import numpy as np
np.random.seed(42)
indices = np.random.permutation(len(iris_data))
n_training_samples = 12
learnset_data = iris_data[indices[:-n_training_samples]]
learnset_labels = iris_labels[indices[:-n_training_samples]]
testset_data = iris_data[indices[-n_training_samples:]]
testset_labels = iris_labels[indices[-n_training_samples:]]
print(learnset_data[:4], learnset_labels[:4])
print(testset_data[:4], testset_labels[:4])
[[ 6.1 2.8 4.7 1.2]
[ 5.7 3.8 1.7 0.3]
[ 7.7 2.6 6.9 2.3]
[ 6. 2.9 4.5 1.5]] [1 0 2 1]
[[ 5.7 2.8 4.1 1.3]
[ 6.5 3. 5.5 1.8]
[ 6.3 2.3 4.4 1.3]
[ 6.4 2.9 4.3 1.3]] [1 2 1 1]