def setUp(self): self.config = create_config(self.dataset_name, model=self.model_name) self.config.set_all({"lookup_embedder.dim": 32}) self.config.set_all(self.options) self.dataset_folder = get_dataset_folder(self.dataset_name) self.dataset = Dataset.create( self.config, folder=get_dataset_folder(self.dataset_name) ) self.model = KgeModel.create(self.config, self.dataset)
def setUp(self): self.config = create_config(self.dataset_name) self.config.set_all({"lookup_embedder.dim": 32}) self.config.set("job.type", "train") self.config.set("train.type", self.train_type) self.config.set_all(self.options) self.dataset_folder = get_dataset_folder(self.dataset_name) self.dataset = Dataset.create(self.config, folder=get_dataset_folder( self.dataset_name)) self.model = KgeModel.create(self.config, self.dataset)
def setUp(self) -> None: self.dataset_name = "dataset_test" self.dataset_folder = get_dataset_folder(self.dataset_name) self.config = create_config(self.dataset_name) self.remove_pickle_files()