Spaces:
Runtime error
Runtime error
File size: 469 Bytes
01f65eb |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
import os
import torch
import pickle
import numpy as np
from PIL import Image, ImageDraw, ImageFont
from ..utils.utilities import Utility
class GoEmotionsDataset(torch.utils.data.Dataset):
util = Utility()
def __init__(self, embeddings, labels):
self.labels = labels
self.instances = embeddings
def __len__(self):
return self.instances.shape[0]
def __getitem__(self, idx):
return self.instances[idx], self.labels[idx] |