Step 5: Make Predictions

When your session has completed, you have your first federated model! You can test it out by making predictions. You can use test.parquet for this step.

This command is going to predict the target column and save the whole dataset with a filled target column as an output.

hfl predict --token $TOKEN --session-id <session_id> --dataset-path <data_path> --batch-size 10

1. <session_id> to reference the completed session

2. <data_path> is the path of the data to make predictions on. Please use the absolute path of your dataset. eg: /root/demo/test.parquet.

The prediction output will be saved as file under the same directory as the input data with the file name <original_input_file_name>_<timestamp>.