Learn practical skills, build real-world projects, and advance your career
Updated 4 years ago
import torch
import torch.nn as nn
import torch.nn.functional as F
from torch.utils.data import DataLoader, TensorDataset, random_split
import pandas as pd
import matplotlib.pyplot as plt
import numpy as np
dataset = pd.read_csv('datasets_19_420_Iris.csv')
target_data = dataset[['Species']]
input_data = dataset.drop(['Id','Species'],axis=1)
print(input_data)
SepalLengthCm SepalWidthCm PetalLengthCm PetalWidthCm
0 5.1 3.5 1.4 0.2
1 4.9 3.0 1.4 0.2
2 4.7 3.2 1.3 0.2
3 4.6 3.1 1.5 0.2
4 5.0 3.6 1.4 0.2
.. ... ... ... ...
145 6.7 3.0 5.2 2.3
146 6.3 2.5 5.0 1.9
147 6.5 3.0 5.2 2.0
148 6.2 3.4 5.4 2.3
149 5.9 3.0 5.1 1.8
[150 rows x 4 columns]
target_data = pd.get_dummies(target_data.Species)
target_data.head()