This script is an exercise to apply neural networks to a classification problem using PyTorch. The dataset (GSE6008, taken from https://sbcb.inf.ufrgs.br/cumida) contains genetic markers (>20000) associated with particular types of ovarian cancer. The script applies PCA to reduce the number of features and then trains a neural network with two hidden layers in order to predict the cancer type. It uses tensorboard to log the losses during the training and then computes the accuracy of the network's productions on the test data.
-
Notifications
You must be signed in to change notification settings - Fork 0
dimashenme/cancer-exercise
About
No description, website, or topics provided.
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published