Basic functions in Pytorch

#1

I have been able to do so much so far, would really appreciate helps from others to do remaining tasks.
Thank you :slight_smile:

from torch.utils.data import Dataset

import torch
import torchvision
import pandas as pd
from skimage.io import imread
from skimage.color import gray2rgb

class ChallengeDataset(Dataset):
def init(self, dataset,csv_path,train_split=0.8 ):
self.data=pd.read_csv(csv_path)

def getitem (self,index):
pixel_sequence = self.data[‘pixels’][index]
face = [int(pixel) for pixel in pixel_sequence.split(’ ')]
label = self.labels[index]

    return face, label
#2

Is this some kind of assignment you should be working on? :slight_smile:
If that’s the case, I would recommend to use available resources (the docs, this forum etc.) to solve the remaining tasks.