Icon

JKISeason 4-6 - Heart Failure Prediction

<p><strong>Challenge 6: Heart Failure Prediction</strong></p><p><strong>Level: </strong>Medium</p><p><strong>Description:&nbsp;</strong>You are a medical researcher working with a hospital to uncover key risk factors behind heart failure. Using an unbalanced dataset of patient records, your task is to build a predictive model to identify potential heart disease cases. But accuracy alone is not enough—clinicians want to understand <em>why</em> the model makes the predictions it does. Train your model, then apply explainable AI (xAI) techniques to reveal the top three features influencing its decisions. Can your insights help doctors detect heart failure earlier and more effectively?<br><br><em>Beginner-friendly objectives:</em> 1. Load and preprocess the heart disease dataset, ensuring that the data is clean and ready for analysis. 2. Perform a train-test split on the dataset, maintaining the class distribution for accurate model evaluation.<br><br><em>Intermediate-friendly objectives:</em> 1. Implement a parameter optimization loop to fine-tune the model's hyperparameters for improved performance. 2. Within the Parameter Optimization Loop, conduct cross-validation to assess the model's robustness and generalization (default: Naïve Bayes, but feel free to experiment with other models). 3. Integrate multiple data science techniques, including one-hot encoding and normalization, to enhance the model's predictive power. 4. Evaluate the model's performance using advanced metrics and visualization techniques to gain insights into its accuracy and reliability. 5. Use the Surrogate Random Forest model from Global Feature Importance to determine the top 3 most important features driving predictions.<br><br>What are the top 3 features responsible for the model's predictions?<br><br><strong>Author</strong>: <strong>Keerthan Shetty</strong><br><br><strong>Dataset:</strong> <strong>Heart Failure Dataset in KNIME Community Hub</strong><br><br>Remember to upload your solution with tag&nbsp;<strong>JKISeason4-6</strong>&nbsp;to your public space on KNIME Community Hub. To increase the visibility of your solution, also post it to&nbsp;<strong>this challenge thread</strong>&nbsp;on KNIME Forum.</p>

URL: Just KNIME It! https://www.knime.com/just-knime-it

Nodes

Extensions

Links