import numpy as np

class Dataset(object):
    def __init__(self, data, labels, ids):
        self.data = data
        self.labels = labels
        self.ids = ids

    def size(self):
        return len(self.data)

    def nclasses(self):
        return len(np.unique(self.labels))

    def sample(self, sid):
        return (self.data[sid], self.labels[sid], self.ids)

0807a9bd06cc04a78846c76553a05aaa722120d8
<mask token>


class Dataset(object):

    def __init__(self, data, labels, ids):
        self.data = data
        self.labels = labels
        self.ids = ids
    <mask token>

    def nclasses(self):
        return len(np.unique(self.labels))
    <mask token>

<mask token>


class Dataset(object):

    def __init__(self, data, labels, ids):
        self.data = data
        self.labels = labels
        self.ids = ids
    <mask token>

    def nclasses(self):
        return len(np.unique(self.labels))

    def sample(self, sid):
        return self.data[sid], self.labels[sid], self.ids

<mask token>


class Dataset(object):

    def __init__(self, data, labels, ids):
        self.data = data
        self.labels = labels
        self.ids = ids

    def size(self):
        return len(self.data)

    def nclasses(self):
        return len(np.unique(self.labels))

    def sample(self, sid):
        return self.data[sid], self.labels[sid], self.ids

import numpy as np


class Dataset(object):

    def __init__(self, data, labels, ids):
        self.data = data
        self.labels = labels
        self.ids = ids

    def size(self):
        return len(self.data)

    def nclasses(self):
        return len(np.unique(self.labels))

    def sample(self, sid):
        return self.data[sid], self.labels[sid], self.ids

import numpy as np

class Dataset(object):
    def __init__(self, data, labels, ids):
        self.data = data
        self.labels = labels
        self.ids = ids

    def size(self):
        return len(self.data)

    def nclasses(self):
        return len(np.unique(self.labels))

    def sample(self, sid):
        return (self.data[sid], self.labels[sid], self.ids)
