application_vnd.jupyter.stderr_output_1.png
import pandas as pd

df_train = pd.read_csv('/kaggle/input/amazon-reviews/train.csv')
df_test = pd.read_csv('/kaggle/input/amazon-reviews/test.csv')

df_train.columns = ['label', 'title', 'text']

df_test.columns = ['label', 'title', 'text']

df_train = df_train.head(100000)
df_test = df_test.head(10000)

print(df_train['label'].value_counts())
print(df_test['label'].value_counts())