ROUNDS = 50
CATEGORIES = ['EducationalInstitution', 'Artist', 'Company', 'MeanOfTransportation', 'OfficeHolder']
TEST_SET_SIZE_PER_CLASS = 1_000
WORKER_SET_SIZE_PER_CLASS = 4_000
MODEL_DIR = '../models_trained'
