Module pipelines.utils.predict_flow.tasks

Tasks for the predict flow.

Functions

def generate_dataframe_from_predictions(predictions: List[Any], output_column_name: str, include_timestamp: bool = False, timestamp: str = None, save_path: Union[str, pathlib.Path] = None) ‑> pandas.core.frame.DataFrame

Generate a dataframe from the predictions.

def get_model(model_name: str, model_version_or_stage: str, tracking_server_uri: str = None)

Get model from MLflow model registry.

def predict(data: Dict[str, List[Any]], model: mlflow.pyfunc.PyFuncModel) ‑> numpy.ndarray

Uses an MLflow model to predict using the data in the dataframe.

def prepare_dataframe_for_prediction(dataframe: pandas.core.frame.DataFrame) ‑> Dict[str, List[Any]]

Use pandas split