Source code for pimlico.datatypes.formatters.features
# This file is part of Pimlico
# Copyright (C) 2016 Mark Granroth-Wilding
# Licensed under the GNU GPL v3.0 - http://www.gnu.org/licenses/gpl-3.0.en.html
from pimlico.cli.browser.formatter import DocumentBrowserFormatter
from pimlico.datatypes import FeatureListScoreDocumentType
[docs]class FeatureListScoreFormatter(DocumentBrowserFormatter):
DATATYPE = FeatureListScoreDocumentType
def __init__(self, corpus):
super(FeatureListScoreFormatter, self).__init__(corpus)
self.feature_list = corpus.data_point_type_instance.features
[docs] def format_document(self, doc):
return u"\n".join(
u"%s: %s" % (
score,
u", ".join(
self.feature_list[f] if weight == 1 else u"%s (%s)" % (self.feature_list[f], weight)
for (f, weight) in features
)
) for (score, features) in doc
)