sagemaker.core.jumpstart.search

sagemaker.core.jumpstart.search#

Functions

search_public_hub_models(query[, hub_name, ...])

Search and filter models from hub using a keyword expression.
sagemaker.core.jumpstart.search.search_public_hub_models(query: str, hub_name: str | None = 'SageMakerPublicHub', sagemaker_session: Session | None = None) → List[HubContent][source]#

Search and filter models from hub using a keyword expression.

Parameters:

  • query (str) – A logical expression used to filter models by keywords. Example: “@task:text-generation AND NOT @framework:legacy”

  • hub_name (Optional[str]) – The name of the hub to query. Defaults to “SageMakerPublicHub”.

  • sagemaker_session (Optional[Session]) – An optional SageMaker Session object. If not provided, a default session will be created and a warning will be logged.

Returns:

A list of filtered HubContent model objects that match the query.

Return type:

List[HubContent]

Contents