Skip to main content

classy.scripts.model.predict

Functions

file_main

def file_main(
    model_checkpoint_path: str,
    input_path: str,
    output_path: str,
    prediction_params: str,
    cuda_device: int,
    token_batch_size: int,
)

interactive_main

def interactive_main(
    model_checkpoint_path: str,
    prediction_params: str,
    cuda_device: int,
)

main

def main()

parse_args

def parse_args()