Use LightGBM to predict the survival of Titanic passengers.

[Instructions]

1_train.ipynb

Set the parameters:

• train_input_filename: input data file path

• model_filename: the path of the output prediction file

• scaler_filename: output normalized file path

• train_output_filename: the path of the output prediction file

• numIteration: the number of Iterations

After the setting is completed, it can be executed.

2. 2_inference.ipynb

Set the parameters:

• inference_input_filename: input data file path

• model_filename: input model file path

• scaler_filename: input normalized file path

• inference_output_filename: the path of the output prediction file

After the setting is completed, it can be executed.

After execution, you can see the result of LightGBM inference.

Data-LightGBM-Classification-inference-Jupyter.png

This SDK is built in AppForAI - AI Dev Tools.

Purchase license separately: USD 600, permanent authorization, single APP authorization, single machine authorization, one-year activation, one-year download, one-year update, one-year email technical support.