-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathIris_flower_classifier
More file actions
34 lines (26 loc) · 1.01 KB
/
Iris_flower_classifier
File metadata and controls
34 lines (26 loc) · 1.01 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
# Importing necessary libraries
from sklearn.datasets import load_iris
from sklearn.model_selection import train_test_split
from sklearn.tree import DecisionTreeClassifier
from sklearn.metrics import accuracy_score
# Loading the Iris dataset
iris = load_iris()
X = iris.data # Features
y = iris.target # Labels
# Split data into training (80%) and testing (20%) sets
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)
# Train a Decision Tree model
model = DecisionTreeClassifier()
model.fit(X_train, y_train)
# Make predictions
y_pred = model.predict(X_test)
# Calculate accuracy
accuracy = accuracy_score(y_test, y_pred)
print(f"Model Accuracy: {accuracy:.2f}")
# Test with a new sample (Example input)
samples = [[5.1, 3.5, 1.4, 0.2],
[6.0, 2.9, 4.5, 1.5],
[7.2, 3.0, 5.8, 1.6]] # List of multiple samples
predictions = model.predict(samples)
for i, pred in enumerate(predictions):
print(f"Sample {i+1}: Predicted Species -", iris.target_names[pred])