# Path: gbdt/_libgbdt.py



from ._libgbdt import libgbdt
            import pandas
            import numpy
            import pandas


class DataStore:
    def __init__(self, data_store):
        self._data_store = data_store

    def __len__(self):
        return len(self._data_store)

    def _get_col(self, k):
        if k not in self._data_store:
            raise ValueError("Column '{}' cannot be found.".format(k))
        type = self._data_store.get_column_type(k)

        if type == 'bucketized_float':
            return self._data_store.get_bucketized_float_col(k)
        elif type == 'string':
            return self._data_store.get_string_col(k)
        else:

# Path: gbdt/_libgbdt.py



# Path: gbdt/_data_store.py
from ._libgbdt import libgbdt
            import pandas
            import numpy
            import pandas



class DataStore:
    def __init__(self, data_store):
        self._data_store = data_store

    def __len__(self):
        return len(self._data_store)

    def _get_col(self, k):
        if k not in self._data_store:
            raise ValueError("Column '{}' cannot be found.".format(k))
        type = self._data_store.get_column_type(k)

        if type == 'bucketized_float':
            return self._data_store.get_bucketized_float_col(k)
        elif type == 'string':
            return self._data_store.get_string_col(k)
        else:

            return self._data_store.get_raw_float_col(k)