Skip to content

doolingdavid/newbreastcancer

Repository files navigation

breastcancer

This repository contains the source code for a machine learning algorithm using scikit-learn 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 BREASTCLASSIFIER folder is a sklearn RandomForestClassifier object that implements the transformation of censore data to a form appropriate for machine learning algorithms as described here:

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

breastcancer

About

Breast Cancer Survival Curve prediction, the RF parameters: n_estimators = 20, max_depth = 15, max_features = .5

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published