def _check_features(table_schema):
    """
    Get list of features to check based on table schema

    Parameters
    ----------
    table_schema: table schema

    Returns
    -------
    exclude_features:
        list of features to exclude during the checking process
    check_features:
        a dictionary of (data type, feature list) to check
    """
    # exclude some features where include equals to 0
    exclude_features = list(table_schema[table_schema['include'] == 0]['column'].values)

    # classify features based on data type
    table_schema_include = table_schema[table_schema['include'] == 1].reset_index(drop=True)
    correct_types = ['key', 'numeric', 'str', 'date']
    check_features = {}
    for correct_type in correct_types:
        check_features[correct_type] = table_schema_include[table_schema_include['type'] == correct_type]['column'].values

    return exclude_features, check_features