query_strategy.py 文件源码

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)