text_plain_output_1.png
import glob

import pandas as pd

config = {'SEED': 42, 'DEBUG': False, 'test_size': 0.1, 'img_size': 128, 'batch_size': 32, 'num_labels': 0, 'epochs': 10, 'device': 'GPU'}
debug_config = {}

df_train = pd.read_csv('../input/kaggle-pog-series-s01e02/train.csv')
df_test = pd.read_csv('../input/kaggle-pog-series-s01e02/test.csv')
df_train['ID'] = df_train['filename'].str.split('.').str[0]
df_test['ID'] = df_test['filename'].str.split('.').str[0]
train_files = pd.DataFrame(glob.glob('../input/pog-train-and-test-spectograms/kaggle/spectograms/train/*.jpg'), columns=['spec_path'])
test_files = pd.DataFrame(glob.glob('../input/pog-train-and-test-spectograms/kaggle/spectograms/test/*.jpg'), columns=['spec_path'])
train_files['ID'] = train_files['spec_path'].str.split('/').str[-1].str.split('.').str[0]
test_files['ID'] = test_files['spec_path'].str.split('/').str[-1].str.split('.').str[0]
df_train_spec = pd.merge(df_train, train_files, how='right', on='ID')
df_test_spec = pd.merge(df_test, test_files, how='right', on='ID')

config['num_labels'] = df_train_spec['genre_id'].nunique()
df_train_spec.genre_id.value_counts(normalize=True) * 100