Pular para conteúdo

Export models

export_beta_geo_fitter(bgf)

Exports the BetaGeoFitter object to a pickle file.

Parameters:

Name Type Description Default
bgf BetaGeoFitter

The BetaGeoFitter object to be exported

 required

Returns:

Type Description
None

None
Source code in api/utils/export_models.py 
10
11
12
13
14
15
16
17
18
19
20
21
def export_beta_geo_fitter(bgf: BetaGeoFitter) -> None:
    """
    Exports the BetaGeoFitter object to a pickle file.

    Args:
        bgf (BetaGeoFitter): The BetaGeoFitter object to be exported

    Returns:
        None
    """
    with open(f"{PROJECT_PATH}/models/beta_geo_fitter.pkl", "wb") as file:
        cloudpickle.dump(bgf, file)

export_gamma_gamma_fitter(ggf)

Exports the GammaGammaFitter object to a pickle file.

Parameters:

Name Type Description Default
ggf GammaGammaFitter

The GammaGammaFitter object to be exported.

 required

Returns:

Type Description
None

None
Source code in api/utils/export_models.py 
24
25
26
27
28
29
30
31
32
33
34
35
def export_gamma_gamma_fitter(ggf: GammaGammaFitter) -> None:
    """
    Exports the GammaGammaFitter object to a pickle file.

    Args:
        ggf (GammaGammaFitter): The GammaGammaFitter object to be exported.

    Returns:
        None
    """
    with open(f"{PROJECT_PATH}/models/gamma_gamma_fitter.pkl", "wb") as file:
        cloudpickle.dump(ggf, file)