lerot.query

Interface to query data with functionality for reading queries from svmlight format, both sequentially and in batch mode.

class lerot.query.Query(qid, feature_vectors, labels=None, comments=None)[source]
get_comment(docid)[source]
get_comments()[source]
get_docids()[source]
get_document_count()[source]
get_feature_vector(docid)[source]
get_feature_vectors()[source]
get_ideal()[source]
get_label(docid)[source]
get_labels()[source]
get_prediction(docid)[source]
get_predictions()[source]
get_qid()[source]
has_ideal()[source]
set_feature_vector(docid, feature_vector)[source]
set_ideal(ideal)[source]
set_label(docid, label)[source]
set_labels(labels)[source]
set_predictions(predictions)[source]
write_to(fh, sparse=False)[source]
class lerot.query.Queries(fh, num_features, preserve_comments=False)[source]

a list of queries with some convenience functions

get_feature_vectors()[source]
get_labels()[source]
get_predictions()[source]
get_qids()[source]
get_query(index)[source]
get_size()[source]
keys()[source]
set_predictions()[source]
values()[source]
class lerot.query.QueryStream(fh, num_features, preserve_comments=False)[source]

iterate over a stream of queries, only keeping one query at a time

next()[source]
read_all()[source]
lerot.query.load_queries(filename, features, preserve_comments=False)[source]

Utility method for loading queries from a file.

lerot.query.write_queries(filename, queries)[source]

Utility method for writing queries to a file. Returns the number of queries written