predictors#

class aiearth.predict.predictors.ImageEchoPredictor(checkpoint: ModelCheckpoint)[源代码]#

基类：Predictor

init(checkpoint: ModelCheckpoint)[源代码]#
predict_udf(data: ndarray)[源代码]#
class aiearth.predict.predictors.ONNXRuntimePredictor(checkpoint: ModelCheckpoint)[源代码]#

基类：Predictor

classmethod from_checkpoint(checkpoint: ModelCheckpoint)[源代码]#

Create a specific predictor from a checkpoint.

参数:

  • checkpoint – Checkpoint to load predictor data from.

  • kwargs – Arguments specific to predictor implementations.

返回:

Predictor object.

返回类型:

Predictor

predict_udf(data: ndarray)[源代码]#
class aiearth.predict.predictors.Predictor(checkpoint: ModelCheckpoint)[源代码]#

基类：Predictor

classmethod from_checkpoint(checkpoint: ModelCheckpoint)[源代码]#

Create a specific predictor from a checkpoint.

参数:

  • checkpoint – Checkpoint to load predictor data from.

  • kwargs – Arguments specific to predictor implementations.

返回:

Predictor object.

返回类型:

Predictor

abstract init(checkpoint: ModelCheckpoint)[源代码]#
abstract predict_udf(data: ndarray | Dict[str, ndarray])[源代码]#
class aiearth.predict.predictors.TensorrtPredictor(checkpoint: ModelCheckpoint)[源代码]#

基类：Predictor

classmethod from_checkpoint(checkpoint: ModelCheckpoint)[源代码]#

Create a specific predictor from a checkpoint.

参数:

  • checkpoint – Checkpoint to load predictor data from.

  • kwargs – Arguments specific to predictor implementations.

返回:

Predictor object.

返回类型:

Predictor

predict_udf(data: ndarray)[源代码]#