nip.utils.language_model_database.LanguageModelDbEntry

class nip.utils.language_model_database.LanguageModelDbEntry(model_series: Annotated[str, 'Model Series'], model_name: Annotated[str, 'Model Name'], developer: Annotated[str, 'Developer'], uri: Annotated[str, 'URI'], num_parameters: Annotated[float | None, 'Parameters (10E+9)'] = None, training_flops: Annotated[float | None, 'FLOPs (10E+23)'] = None, mmlu_pro_score: Annotated[float | None, 'MMLU-Pro'] = None, openrouter_input_cost: Annotated[float | None, 'OpenRouter Input Cost'] = None, openrouter_output_cost: Annotated[float | None, 'OpenRouter Output Cost'] = None)

An entry in the language model database.

Methods Summary

__eq__(other)	Return self==value.
__init__(model_series, model_name, ...[, ...])
__repr__()	Return repr(self).
from_row(row)	Create a LanguageModelDbEntry from a database Pandas Series.

Attributes

display_name	A nice name for the model.
mmlu_pro_score
model_id	The ID of the model in the model provider.
num_parameters
openrouter_input_cost
openrouter_output_cost
provider	The service which provides the model API.
training_flops
model_series
model_name
developer
uri

Methods

__eq__(other)

Return self==value.

__init__(model_series: Annotated[str, 'Model Series'], model_name: Annotated[str, 'Model Name'], developer: Annotated[str, 'Developer'], uri: Annotated[str, 'URI'], num_parameters: Annotated[float | None, 'Parameters (10E+9)'] = None, training_flops: Annotated[float | None, 'FLOPs (10E+23)'] = None, mmlu_pro_score: Annotated[float | None, 'MMLU-Pro'] = None, openrouter_input_cost: Annotated[float | None, 'OpenRouter Input Cost'] = None, openrouter_output_cost: Annotated[float | None, 'OpenRouter Output Cost'] = None) → None

__repr__()

Return repr(self).

classmethod from_row(row: Series) → LanguageModelDbEntry

Create a LanguageModelDbEntry from a database Pandas Series.

Parameters:

row (pd.Series) – The row of the Pandas DataFrame containing the model information

Returns:

LanguageModelDbEntry – The LanguageModelDbEntry object created from the row