import numpy as np
from sklearn.datasets import load_iris
from sklearn.decomposition import PCA
# import matplotlib as mpl
# import matplotlib.pyplot as plt
RANDOM_STATE_SEED = 123
np.random.seed(RANDOM_STATE_SEED)
iris = load_iris()
X_raw = iris['data']
y_raw = iris['target']
# For visualization purposes, we apply PCA to the original dataset
pca = PCA(n_components=2,random_state=RANDOM_STATE_SEED)
transformed_iris = pca.fit_transform(X=X_raw)
#Isolate our example for our labeled dataset
n_labeled_examples = X_raw.shape[0]
training_indices = np.random.randint(low=0,high=n_labeled_examples+1,size=3)
X_train = X_raw[training_indices]
y_train = X_raw[training_indices]
# Isolate the non-training examples we'll be querying
X_pool = np.delete(X_raw,training_indices,axis=0)
y_pool = np.delete(y_raw,training_indices,axis=0)