Dataset

Complexity

Total Complexity

5

Size/Duplication

Total Lines	32
Duplicated Lines	0 %

Importance

Changes	1
Bugs	0	Features	0

4 Methods

Rating	Name	Duplication	Size	Complexity
A	train_size()	0	10	2
A	test_set()	0	4	1
A	train_set()	0	5	1
A	valid_set()	0	4	1

#!/usr/bin/env python
# -*- coding: utf-8 -*-

from abc import ABCMeta, abstractmethod
import collections

class Dataset(object):
    """
    Abstract dataset class.
    """
    __metaclass__ = ABCMeta

    @abstractmethod
    def train_set(self):
        """
        :rtype: list of tuple
        """

    def valid_set(self):
        """
        :rtype: list of tuple
        """

    def test_set(self):
        """
        :rtype: list of tuple
        """

    def train_size(self):
        """
        Return size of training data. (optional)
        :rtype: number
        """
        train_set = self.train_set()
        if isinstance(train_set, collections.Iterable):
            return len(list(train_set))
        else:
            return None