ml002.py

# =========================================================
#
# From: www.youtube.com/watch?v=tNa99PG8HR8

# Flower data set:
#     en.wikipedia.org/wiki/Iris_flower_data_set
# =========================================================
# Goal:
# 1. Import dataset.
# 2. Train classifier.
# 3. Predict label for new flower.
# 4. Visualize the tree.
# =========================================================

from sklearn.datasets import load_iris

iris = load_iris()

print(iris.feature_names)
print(iris.target_names)

print(iris.data[0])
print(iris.target[0])