Слияние кода завершено, страница обновится автоматически
from torch.utils.data import Dataset
import torch
class MyDataset(Dataset):
"""
"""
def __init__(self, data_list):
self.data_list = data_list
def __getitem__(self, index):
input_ids = self.data_list[index].strip()
input_ids = [int(token_id) for token_id in input_ids.split()]
return input_ids
def __len__(self):
return len(self.data_list)
Вы можете оставить комментарий после Вход в систему
Неприемлемый контент может быть отображен здесь и не будет показан на странице. Вы можете проверить и изменить его с помощью соответствующей функции редактирования.
Если вы подтверждаете, что содержание не содержит непристойной лексики/перенаправления на рекламу/насилия/вульгарной порнографии/нарушений/пиратства/ложного/незначительного или незаконного контента, связанного с национальными законами и предписаниями, вы можете нажать «Отправить» для подачи апелляции, и мы обработаем ее как можно скорее.
Опубликовать ( 0 )