library("randomForest");

fileName <- "../data/LungCancerDataset_AllRecords_NORM_reduced_features.csv"

lung_cancer_datatable <- read.csv(fileName, header = TRUE, sep =",", stringsAsFactors = FALSE);

# lung_cancer_datatable$"Typeoffollow.upexpected" <- NULL

# lung_cancer_datatable$"SiterecwithKaposiandmesothelioma" <- NULL

# 

# lung_cancer_datatable$Metastasis <- lung_cancer_datatable$M

# lung_cancer_datatable$M <- NULL

rf_output <- randomForest(Metastasis ~ ., data=lung_cancer_datatable, importance=TRUE, proximity=TRUE)

dd <- as.data.frame(rf_output$importance);

dd_sorted <- dd[order(dd$"%IncMSE"), ]

print(dd_sorted);

varImpPlot(rf_output)