/
spirzen
/
it-code-examples
Обзор
Документация
Войти
/
spirzen
/
it-code-examples
Код
Запросы
0
Задачи
Вики
Пакеты
0
Релизы
0
Аналитика
Безопасность
master
examples/python/py-502-333-003/main.py
20 строк
712 B
Spirzen
Code migration pack
09 июн 2026, 01:41
09 июн 2026, 01:41
cebc284
Код
Авторство
О чём код?
from torch.utils.data import Dataset, DataLoader, random_split class TabularDataset(Dataset): def __init__(self, features, labels): self.x = torch.tensor(features, dtype=torch.float32) self.y = torch.tensor(labels, dtype=torch.long) def __len__(self): return len(self.y) def __getitem__(self, idx): return self.x[idx], self.y[idx] full_dataset = TabularDataset(X, y) train_size = int(0.8 * len(full_dataset)) val_size = len(full_dataset) - train_size train_ds, val_ds = random_split(full_dataset, [train_size, val_size]) train_loader = DataLoader(train_ds, batch_size=64, shuffle=True, num_workers=0) val_loader = DataLoader(val_ds, batch_size=64, shuffle=False)