MCPcopy Create free account

hub / github.com/PersiaML/PERSIA / types & classes

Types & classes217 in github.com/PersiaML/PERSIA

↓ 7 callersClassIDTypeFeature
:class:`IDTypeFeature` is a sparse matrix in `LIL <https://scipy-lectures.org/advanced/scipy_sparse/lil_matrix.html>`_ format which contains categ
persia/embedding/data.py:69
↓ 7 callersClassPersiaBatch
r"""The :class:`PersiaBatch` is the type of dataset used internally in Persia. It wraps the :class:`IDTypeFeature`, :class:`NonIDTypeFeature`, :cl
persia/embedding/data.py:279
↓ 4 callersClassValues
Borrowing iterator over values of the linked array.
rust/persia-embedding-holder/src/array_linked_list.rs:993
↓ 3 callersClassDataCtx
r"""Data context provides the communication functionality to data generator component. Used for sending a :class:`PersiaBatch` to the `nn worker`
persia/ctx.py:274
↓ 3 callersClassDataLoader
examples/src/adult-income/data_generator.py:10
↓ 3 callersClassIndices
rust/persia-embedding-holder/src/array_linked_list.rs:990
↓ 3 callersClassIntoIndexed
rust/persia-embedding-holder/src/array_linked_list.rs:1098
↓ 3 callersClassLabel
:class:`Label` is the ``subclass`` of :class:`NdarrayDataBase` that you can add various datatype and multi-dimensional data. Example for :cla
persia/embedding/data.py:217
↓ 3 callersClassStreamingDataset
r"""Streaming dataset receives the :class:`PersiaBatch` from the upstream data flow sent by :class:`DataCtx`. In the implemented :meth:`.Stre
persia/data.py:97
↓ 2 callersClassAdagrad
rust/persia-common/src/optim.rs:246
↓ 2 callersClassAdam
rust/persia-common/src/optim.rs:99
↓ 2 callersClassBaseCtx
r"""Initializes a common context for other persia context, e.g. :class:`DataCtx`, :class:`EmbeddingCtx` and :class:`TrainCtx`. This class should n
persia/ctx.py:202
↓ 2 callersClassDNN
examples/src/adult-income/model.py:7
↓ 2 callersClassDataLoader
r"""Data loader will preprocess the data to the ``PersiaTrainingBatch``. The :class:`DataLoader` is a pipeline that preprocess the :class:`.Persi
persia/data.py:202
↓ 2 callersClassIDTypeFeatureRemoteRef
rust/persia-common/src/lib.rs:141
↓ 2 callersClassNdarrayDataBase
The :class:`NdarrayDataBase` is a data structure that supports various datatypes and multi-dimensional data. PERSIA needs to convert the :class:`N
persia/embedding/data.py:160
↓ 2 callersClassNonIDTypeFeature
The :class:`NonIDTypeFeature` is the ``subclass`` of :class:`NdarrayDataBase` that you can add various datatypes and multi-dimensional data.
persia/embedding/data.py:256
↓ 2 callersClassSGD
r"""A wrapper to config the embedding-server SGD optimizer.
persia/embedding/optim.py:18
↓ 1 callersClassAdagrad
r"""A wrapper to config the embedding-server Adagrad optimizer.
persia/embedding/optim.py:60
↓ 1 callersClassBackward
rust/persia-core/src/backward.rs:358
↓ 1 callersClassDDPOption
Implements an option to convert torch model to a DDP model. Current backend in :class:`DDPOption` only support `nccl` and `gloo`. You can set
persia/distributed.py:74
↓ 1 callersClassEmbeddingConfig
r"""Embedding hyperparameters, argument of :class:`.EmbeddingCtx`.
persia/embedding/__init__.py:4
↓ 1 callersClassEmbeddingCtx
r"""Provides the embedding-related functionality. :class:`EmbeddingCtx` can run offline test or online inference depending on different preprocess
persia/ctx.py:345
↓ 1 callersClassForward
rust/persia-core/src/forward.rs:834
↓ 1 callersClassGradientBatch
rust/persia-core/src/backward.rs:53
↓ 1 callersClassIDTypeFeatureWithSingleID
The :class:`IDTypeFeatureWithSingleID` is a special format of :class:`IDTypeFeature` where there is only one id for each sample in the batch. :cla
persia/embedding/data.py:116
↓ 1 callersClassInferCtx
r"""Subclass of :class:`EmbeddingCtx` that provides the inference functionality without `nats-servers`. Example for :class:`InferCtx`: .. co
persia/ctx.py:1077
↓ 1 callersClassIntoIndices
rust/persia-embedding-holder/src/array_linked_list.rs:1099
↓ 1 callersClassIntoValues
Owning iterator over values of the linked array.
rust/persia-embedding-holder/src/array_linked_list.rs:1102
↓ 1 callersClassOptimizerBase
rust/persia-core/src/optim.rs:9
↓ 1 callersClassPersiaBatchDataChannel
rust/persia-core/src/utils.rs:102
↓ 1 callersClassPersiaCommonContext
rust/persia-core/src/lib.rs:191
↓ 1 callersClassPersiaServiceCtx
Launch the required processes to mock the distributed PERSIA environment. You can add the embedding config file path or a dict that contains the
persia/helper.py:125
↓ 1 callersClassRawEmbedding
rust/persia-core/src/forward.rs:50
↓ 1 callersClassSumEmbedding
rust/persia-core/src/forward.rs:45
↓ 1 callersClassTestDataset
examples/src/adult-income/train.py:27
↓ 1 callersClassTrainCtx
r"""Subclass of :class:`EmbeddingCtx` that implements a `backward` function to update the embeddings. Example for :class:`TrainCtx`: ..
persia/ctx.py:655
↓ 1 callersClass_Env
persia/env.py:16
↓ 1 callersClasslevelFilter
r"""Log level filter. Arguments: level (int): filter log level. Only logs with level higher than ``level`` will be kept.
persia/logger.py:8
EnumAction
k8s/src/bin/operator.rs:50
ClassAdagradConfig
rust/persia-common/src/optim.rs:33
ClassAdam
r"""A wrapper to config the embedding-server Adam optimizer.
persia/embedding/optim.py:35
ClassAdamConfig
rust/persia-common/src/optim.rs:19
ClassAdamPowerOfBetas
rust/persia-common/src/optim.rs:94
ClassAllEmbeddingServerClient
rust/persia-embedding-server/src/embedding_worker_service/mod.rs:139
InterfaceAllocatable
rust/persia-core/src/cuda/resource_pool.rs:7
ClassApplyRequest
k8s/src/bin/server.rs:24
ClassArrayLinkedList
rust/persia-embedding-holder/src/array_linked_list.rs:206
ClassBackwardImpl
rust/persia-core/src/backward.rs:115
ClassBaguaDistributedOption
Implements an option to convert torch model to a bagua distributed model. Example for :class:`BaguaDistributedOption`: .. code-block:: pytho
persia/distributed.py:259
ClassBatchEmbedding
rust/others/persia-common-benchmark/benches/serialize_inf_request.rs:22
ClassBatchEmbeddingSmallVec
rust/others/persia-common-benchmark/benches/serialize_inf_request.rs:108
ClassBatchEmbeddingsRequest
rust/others/persia-common-benchmark/benches/serialize_inf_request.rs:15
ClassBatchEmbeddingsRequestSmallVec
rust/others/persia-common-benchmark/benches/serialize_inf_request.rs:101
ClassBoundedGammaInitialization
rust/persia-embedding-config/src/lib.rs:41
ClassBoundedNormalInitialization
rust/persia-embedding-config/src/lib.rs:66
ClassBoundedPoissonInitialization
rust/persia-embedding-config/src/lib.rs:54
ClassBoundedUniformInitialization
rust/persia-embedding-config/src/lib.rs:28
EnumCPUStorage
rust/persia-core/src/tensor.rs:98
ClassChannelPair
rust/persia-common/src/utils.rs:4
ClassCheckpointingConfig
rust/persia-embedding-config/src/lib.rs:344
ClassCli
k8s/src/bin/server.rs:196
ClassCli
k8s/src/bin/gencrd.rs:9
ClassCli
rust/persia-embedding-server/src/bin/persia-embedding-parameter-server.rs:26
ClassCli
rust/persia-embedding-server/src/bin/persia-embedding-worker.rs:26
ClassCompactBatchEmbedding
rust/others/persia-common-benchmark/benches/serialize_inf_request.rs:65
ClassCompactBatchEmbeddingsRequest
rust/others/persia-common-benchmark/benches/serialize_inf_request.rs:57
ClassCompactBatchEmbeddingsRequestWithoutIndices
rust/others/persia-common-benchmark/benches/serialize_inf_request.rs:252
ClassContextData
k8s/src/bin/operator.rs:40
ClassCudaEventPtr
rust/persia-core/src/cuda/cuda_event_pool.rs:12
ClassCudaMallocPtr
rust/persia-core/src/cuda/cuda_memory_pool.rs:11
ClassCudaStreamPtr
rust/persia-core/src/cuda/cuda_stream_pool.rs:10
ClassDLDataType
rust/persia-core/src/dlpack.rs:45
EnumDLDataTypeCode
rust/persia-core/src/dlpack.rs:30
ClassDLDevice
rust/persia-core/src/dlpack.rs:54
EnumDLDeviceType
rust/persia-core/src/dlpack.rs:11
ClassDLManagedTensor
rust/persia-core/src/dlpack.rs:81
ClassDLTensor
rust/persia-core/src/dlpack.rs:62
EnumDTypeImpl
rust/persia-core/src/tensor.rs:24
ClassDataLoaderSpec
k8s/src/crd.rs:503
ClassDataflowService
rust/persia-core/src/nats.rs:117
ClassDevice
rust/persia-core/src/tensor.rs:177
EnumDeviceType
rust/persia-core/src/tensor.rs:171
ClassDistributedBaseOption
Implements a common option to convert torch model to a distributed data parallel model, e.g. Bagua Distributed or pyTorch DDP. This class sho
persia/distributed.py:13
ClassDtype
rust/persia-core/src/forward.rs:102
ClassEmbedding
rust/persia-core/src/forward.rs:58
ClassEmbeddingBackwardPacket
rust/persia-core/src/backward.rs:108
ClassEmbeddingBatch
rust/persia-common/src/lib.rs:110
ClassEmbeddingConfig
rust/persia-embedding-config/src/lib.rs:554
ClassEmbeddingGradientBatch
rust/persia-common/src/grad.rs:45
EnumEmbeddingImpl
rust/persia-core/src/forward.rs:39
ClassEmbeddingModelInfo
rust/persia-model-manager/src/lib.rs:57
ClassEmbeddingModelManager
rust/persia-model-manager/src/lib.rs:74
EnumEmbeddingModelManagerError
rust/persia-model-manager/src/lib.rs:27
EnumEmbeddingModelManagerStatus
rust/persia-model-manager/src/lib.rs:64
ClassEmbeddingMonitorInner
rust/persia-embedding-server/src/monitor.rs:29
ClassEmbeddingParameterNatsService
rust/persia-embedding-server/src/embedding_parameter_service/mod.rs:596
ClassEmbeddingParameterServerConfig
rust/persia-embedding-config/src/lib.rs:419
EnumEmbeddingParameterServerError
rust/persia-embedding-server/src/embedding_parameter_service/mod.rs:82
ClassEmbeddingParameterServerSpec
k8s/src/crd.rs:472
next →1–100 of 217, ranked by callers