Challenge: Preparing the Iris Dataset
Завдання
Swipe to start coding
- Create a
TensorDataset
. - Wrap the
TensorDataset
in aDataLoader
with a batch size of32
and shuffling enabled.
Рішення
99
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
import torch
from torch.utils.data import TensorDataset, DataLoader
import pandas as pd
torch.manual_seed(42)
iris_df = pd.read_csv('https://content-media-cdn.codefinity.com/courses/1dd2b0f6-6ec0-40e6-a570-ed0ac2209666/section_2/iris.csv')
features = iris_df.drop(columns='species').values
target = iris_df['species'].values
# Convert features and target into PyTorch tensors
features_tensor = torch.tensor(features, dtype=torch.float32)
target_tensor = torch.tensor(target, dtype=torch.long)
# Create a TensorDataset
iris_dataset = TensorDataset(features_tensor, target_tensor)
# Wrap the dataset in a DataLoader
iris_loader = DataLoader(iris_dataset, batch_size=32, shuffle=True)
# Display the DataLoader in action
for batch_idx, (batch_features, batch_targets) in enumerate(iris_loader):
print(f"Batch {batch_idx + 1}:")
print("Features:\n", batch_features)
print("Targets:\n", batch_targets)
break # Display only the first batch
Все було зрозуміло?
Дякуємо за ваш відгук!
Секція 2. Розділ 6
single
99
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
import torch
from torch.utils.data import TensorDataset, DataLoader
import pandas as pd
torch.manual_seed(42)
iris_df = pd.read_csv('https://content-media-cdn.codefinity.com/courses/1dd2b0f6-6ec0-40e6-a570-ed0ac2209666/section_2/iris.csv')
features = iris_df.drop(columns='species').values
target = iris_df['species'].values
# Convert features and target into PyTorch tensors
features_tensor = torch.tensor(features, dtype=torch.float32)
target_tensor = torch.tensor(target, dtype=torch.long)
# Create a TensorDataset
iris_dataset = ___
# Wrap the dataset in a DataLoader
iris_loader = ___
# Display the DataLoader in action
for batch_idx, (batch_features, batch_targets) in enumerate(iris_loader):
print(f"Batch {batch_idx + 1}:")
print("Features:\n", batch_features)
print("Targets:\n", batch_targets)
break
Запитати АІ
Запитати АІ
Запитайте про що завгодно або спробуйте одне із запропонованих запитань, щоб почати наш чат