def setUp(self):
dataset_filepath = config.ROOT + '/python/test/resource/NFLX_dataset_public_raw.py'
dataset = import_python_file(dataset_filepath)
np.random.seed(0)
info_dict = {
'selected_subjects': range(5),
}
self.dataset_reader = SelectSubjectRawDatasetReader(dataset, input_dict=info_dict)
def run_one_num_subject(num_subject, dataset, seed):
np.random.seed(seed)
total_subject = len(dataset.dis_videos[0]['os'])
info_dict = {
'selected_subjects':
np.random.permutation(total_subject)[:num_subject]
}
dataset_reader = SelectSubjectRawDatasetReader(
dataset, input_dict=info_dict)
subjective_model = model_class(dataset_reader)
result = subjective_model.run_modeling(normalize_final=False)
return dataset_reader, result
