xgbse

Input :

• Import Data: Imports data using the dataset_for_surv_analysis function.

     URI_SQLITE_DB = "data/dnv_coe.db"

• Select Columns and Prepare Data: Necessary columns are selected based on settings.

Process :

• Split Data into Features (X) and Target (y): Split the data into feature and target variables.
• Split Data into Training and Validation Sets: Split the data into training and validation sets.
• Convert Data to XGBoost Format: Convert the data into a format suitable for XGBoost.
• Train the Model: Train the model using the training data.
• Predict Survival Probabilities: Use the trained model to predict survival probabilities on the validation data.
• Evaluate the Model: Evaluate the model's performance using the concordance index.

Outputs

• Model Predictions: The predictions made by the model on the validation data.
• Model Evaluation Metrics: Metrics such as the concordance index to evaluate the model's performance.