ContextFeatureSet

class ContextFeatureSet(**kwargs)

Bases: collections.UserDict, Mapping[str, torch.FloatTensor]

Context feature set for biological/chemical context feature vectors.

Methods Summary

from_dict(data)	Generate a context feature set from a data dictionary.
get_feature_matrix(contexts)	Get the feature matrix for a list of contexts.

Methods Documentation

classmethod from_dict(data)

Generate a context feature set from a data dictionary.

Return type

ContextFeatureSet

get_feature_matrix(contexts)

Get the feature matrix for a list of contexts.

Parameters

contexts (Iterable[str]) – A list of context identifiers.

Return type

FloatTensor

Returns

A matrix of context features.