def __init__(self, dataset, save_path_queries=None, **kwargs):
super(UncertaintyDenseSampling, self).__init__(dataset, **kwargs)
self.model = kwargs.pop('model', None)
if self.model is None:
raise TypeError(
"__init__() missing required keyword-only argument: 'model'"
)
self.save_path_queries = save_path_queries
self.save_path_queries_hdf5 = os.path.join(self.save_path_queries,
os.path.normpath(self.save_path_queries) + ".hdf5")
if os.path.isfile(self.save_path_queries_hdf5):
print "This file already exists %s" % self.save_path_queries_hdf5
quit(0)
self.model.train(self.dataset, first_time=True)
unlabeled_train = self.dataset.get_unlabeled_train_data()["features"]
print "Computing cosine similarities of", unlabeled_train.shape, "by", unlabeled_train.shape
self.similarity_matrix = cosine_similarity(unlabeled_train, unlabeled_train)
评论列表
文章目录