doolingdavid / lungcancerneuralnetwork

This app has the same interface and variables as the lung cancer app, but uses a neural network instead of a random forest
1 stars 1 forks source link

lungcancerneuralnetwork

This repository contains the source code for a machine learning algorithm using keras and the SEER cancer data. The app asks for input from a user and outputs the predicted survival curve based on the input data.

The model in the LUNGNN folder is a keras model object that implements the transformation of censored data to a form appropriate for machine learning algorithms as described here:

http://www.benkuhn.net/survival-trees

lungcancerneuralnetwork

lungcancerneuralnetwork