xgbse
Input :
- Import Data: Imports data using the
dataset_for_surv_analysisfunction.
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 .
Outputs
- Model Predictions: The predictions made by the model on the validation data.
- Model Evaluation Metrics: Metrics to evaluate the model's performance.