Cómo crear un torch.utils.data.Dataset e importarlo en un torch.utils.data.DataLoader.
Me gustaría importar datos de un archivo CSV con torch.utils.data.Dataset
para poder usar torch.utils.data.DataLoader
para manejarlos. Los datos no pertenecen a torchvision y están en mi PC. Parece que no hay solución en Google. Agradecería mucho si pudieras darme algún consejo.
davy.ai
Para importar datos de un archivo CSV con
torch.utils.data.Dataset
, puedes crear una clase de dataset personalizada que lea el archivo CSV y lo convierta en un tensor. Aquí tienes un ejemplo de implementación:En el método
__init__
, leemos el archivo CSV usandonumpy
y lo convertimos en un tensor usandotorch.tensor
. En el método__getitem__
, devolvemos un solo elemento del tensor en el índice especificado. En el método__len__
, devolvemos la longitud del tensor.Una vez que hayas creado tu clase de dataset personalizada, puedes usar
torch.utils.data.DataLoader
para cargar los datos por lotes:En este código de ejemplo, creamos una instancia de
MyDataset
pasando la ruta al archivo CSV. Luego, creamos una instancia deDataLoader
con un tamaño de lote de 32 y mezclamos los datos al azar. Puedes ajustar el tamaño del lote y otros parámetros según sea necesario para tu caso de uso.