src.model.fps_decorated_model

Classes

FPSDecoratedModel(**kwargs)

class src.model.fps_decorated_model.FPSDecoratedModel(**kwargs)

Author:

Alberto M. Esmoris Pena

Decorator for machine learning models that makes the decorated model work on a FPS-based representation of the point cloud.

The FPS Decorated Model (FPSDecoratedModel) constructs a representation of the point cloud, then it calls the model on this representation and, when used for predicting, it propagates the predictions back to the original point cloud (the one from which the representation was built).

See FPSDecoratorTransformer and DecoratedModel.

Variables:

• fps_decorator_spec (dict) – The specification of the FPS transformation defining the decorator.

• fps_decorator (FPSDecoratorTransformer) – The FPS decorator to be applied on input point clouds.

static extract_model_args(spec)

Extract the arguments to initialize/instantiate a FPSDecoratedModel from a key-word specification.

Parameters:

spec – The key-word specification containing the arguments.

Returns:

The arguments to initialize/instantiate a FPSDecoratedModel.

__init__(**kwargs)

Initialization for any instance of type FPSDecoratedModel.

train(pcloud)

Decorate the main training logic to work on the representation. See Model and Model.train().

decorate_pcloud(pcloud)

See DecoratedModel.decorate_pcloud().

propagate(rf_yhat)

See DecoratedModel.propagate().