Predictions

class Predictions(df, factory)[source]

Bases: ABC

Base class for predictions.

Methods Summary

add_membership_columns(**filter_triples)

Add columns indicating whether the triples are known.

exchange_df(df)

Create a copy of the object with its dataframe exchanged.

filter_triples(*triples)

Filter out known triples.

Methods Documentation

Parameters:
add_membership_columns(**filter_triples)[source]

Add columns indicating whether the triples are known.

Return type:

DataFrame

Parameters:

filter_triples (Tuple[str, str, str] | Sequence[Tuple[str, str, str]] | ndarray | LongTensor | CoreTriplesFactory | None) –

exchange_df(df)[source]

Create a copy of the object with its dataframe exchanged.

Return type:

Predictions

Parameters:

df (DataFrame) –

filter_triples(*triples)[source]

Filter out known triples.

Return type:

DataFrame

Parameters:

triples (Tuple[str, str, str] | Sequence[Tuple[str, str, str]] | ndarray | LongTensor | CoreTriplesFactory | None) –