跳到主要内容

load()

Near Deprecation

此操作会将当前 Collection 的数据加载到内存中。

请求语法​

python
load(
partition_names: list[str] | None,
replica_number: int,
timeout: float | None,
)
Notes

此操作为非阻塞操作。您可以调用 utility.wait_for_loading_complete() 来阻塞当前进程。

参数:

  • partition_names (list(str) | None) -

    要加载的当前 Collection 的 Partition。如果未指定,则会加载所有 Partition。

  • timeout (float | None) -

    此操作的超时时长。将其设置为 None 表示当收到任意响应或发生任意错误时,此操作即超时。

返回类型:

NoneType

返回值:

None

异常:

  • MilvusException

    当此操作过程中发生任何错误时,将引发此异常。

Warning

如果您尝试加载未建立索引的 Collection,将会收到 MilvusException。

示例​

python
from pymilvus import Collection, CollectionSchema, FieldSchema, DataType

schema = CollectionSchema([
FieldSchema("id", DataType.INT64, is_primary=True),
FieldSchema("vector", DataType.FLOAT_VECTOR, dim=5)
])

# Create a collection
collection = Collection(
name="test_collection",
schema=schema
)

# Set the index parameters
index_params = {
"index_type": "AUTOINDEX",
"metric_type": "COSINE",
"params": {
"nprobe": 10
}
}

# Create an index on the vector field
collection.create_index(
field_name="vector",
index_params=index_params,
timeout=None
)

# Load the entire collection with one replica of the collection data
collection.load()

# Load the entire collection with two replicas of the collection data
collection.load(
replica_number=2
)

# Load a specific partition with two replicas of the partition data
collection.load(
partition_names=["partitionA"],
replica_number=2
)

以下操作与 load() 相关:

最低 SDK 版本Inherit