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()