def train(self, data): """Trains the model on given data set. :param data: Data set on which to train the model. :type data: :class:`pandas.DataFrame` """ self.init_model() data = tools.first_answers(data) data.sort(['inserted']).apply(self.update, axis=1)
def split_data(cls, data, ratio=0.7): """Classmethod that splits data into training set and test set. :param data: The object containing data. :type data: :class:`pandas.DataFrame`. :param ratio: What portion of data to include in the training set and the test set. :obj:`0.5` means that the data will be distributed equaly. :type ratio: float """ data = tools.first_answers(data) return tools.split_data(data, ratio=ratio)