This code shows simple examples of each ML type: supervised classification, unsupervised clustering, and a basic reinforcement learning simulation.
from sklearn.datasets import load_iris
from sklearn.model_selection import train_test_split
from sklearn.tree import DecisionTreeClassifier
from sklearn.cluster import KMeans
import numpy as np
# Supervised Learning example
iris = load_iris()
X_train, X_test, y_train, y_test = train_test_split(iris.data, iris.target, random_state=42)
clf = DecisionTreeClassifier()
clf.fit(X_train, y_train)
supervised_accuracy = clf.score(X_test, y_test)
# Unsupervised Learning example
kmeans = KMeans(n_clusters=3, random_state=42)
kmeans.fit(iris.data)
clusters = kmeans.labels_
# Reinforcement Learning example (simple simulation)
# Agent learns to choose action 0 or 1 to get reward
rewards = [1, 0] # action 0 gives reward 1, action 1 gives reward 0
actions = []
for _ in range(5):
action = 0 # agent always picks action 0
actions.append(action)
reward = rewards[action]
print(f"Supervised accuracy: {supervised_accuracy:.2f}")
print(f"Unsupervised cluster labels (first 5): {clusters[:5]}")
print(f"Reinforcement actions taken: {actions}")