def load_data(train_path='input/train_master.tsv', test_path='input/sample_submit.tsv'):
    train = pd.read_csv(train_path, delimiter='\t', index_col=False)
    test = pd.read_csv(test_path, delimiter='\t', index_col=False, header=None)
    print('train shape', train.shape)
    print(train.head())
    X_train = train['file_name'].as_matrix()
    y_train = train['category_id'].as_matrix()
    X_test = test.iloc[:, 0]
    return (X_train, y_train, X_test)
def load_data(train_path='input/train_master.tsv', test_path='input/sample_submit.tsv'):
    train = pd.read_csv(train_path, delimiter='\t', index_col=False)
    test = pd.read_csv(test_path, delimiter='\t', index_col=False, header=None)
    print('train shape', train.shape)
    print(train.head())
    X_train = train['file_name'].as_matrix()
    y_train = train['category_id'].as_matrix()
    X_test = test.iloc[:, 0]
    return (X_train, y_train, X_test)