summaryrefslogtreecommitdiff
path: root/scripts/padel-tsne.R
blob: b8e97637bda94f1994b4cdbd02132bc41080f22f (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
#!/usr/bin/env Rscript
library(Rtsne)
library(ggplot2)
data <- read.csv("figures/tsne-padel.csv")
labels <- data$Mutagenicity
data$Mutagenicity <- NULL
m <- as.matrix(data)
tsne <- Rtsne(m,verbose=T,check_duplicates=F)
#write.csv(tsne,"tsne.csv")
write.csv(data.frame(x = tsne$Y[,1], y = tsne$Y[,2]),"padel-tsne.csv")
tsne_plot <- data.frame(x = tsne$Y[,1], y = tsne$Y[,2])
colors <- c("PA" = "#00BFC4", "mutagen" = "#F8766D", "non-mutagen" = "#7CAE00")
plot <- ggplot(tsne_plot)
plot + geom_point(aes(x=x, y=y, color = labels)) + xlab(element_blank()) + ylab(element_blank()) + theme(axis.ticks = element_blank(), axis.text = element_blank(), legend.title=element_blank()) + scale_color_manual(values = colors)
ggsave("figures/tsne-padel.png")