forked from AAAI/AINews
-
Notifications
You must be signed in to change notification settings - Fork 0
/
corpus-mds.r
67 lines (60 loc) · 2.25 KB
/
corpus-mds.r
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
library(ggplot2)
args <- commandArgs(trailingOnly = T)
directory <- args[1]
corpus <- read.csv(paste(directory,"/corpus.csv", sep=""))
fit <- cmdscale(corpus, k=2)
data <- as.data.frame(fit)
data$Category <- gsub("\\d+ ", "", rownames(data))
Category <- factor(gsub("\\d+ ", "", rownames(fit)))
png(paste(directory,"/corpus-mds.png", sep=""), width=500, height=500, res=100)
p <- ggplot(data) +
geom_point(aes(x=V1, y=V2, color=Category)) +
scale_x_continuous("", breaks=NA) +
scale_y_continuous("", breaks=NA) +
opts(axis.text.x = theme_blank(), axis.title.x=theme_blank(),
axis.text.y = theme_blank(), axis.title.y=theme_blank(),
legend.position = "none")
p
dev.off()
corpus <- read.csv(paste(directory,"/models.csv", sep=""))
fit <- cmdscale(corpus, k=2)
data <- as.data.frame(fit)
data$Category <- rownames(data)
png(paste(directory,"/corpus-mds-centroids.png", sep=""),
width=500, height=500, res=100)
p <- ggplot(data) +
geom_text(aes(x=V1, y=V2, label=Category, size=3, color=Category)) +
scale_x_continuous("", breaks=NA) +
scale_y_continuous("", breaks=NA) +
opts(axis.text.x = theme_blank(), axis.title.x=theme_blank(),
axis.text.y = theme_blank(), axis.title.y=theme_blank(),
legend.position = "none")
p
dev.off()
png(paste(directory,"/corpus-mds-faceted.png", sep=""),
width=500, height=500, res=100)
p <- ggplot(data)
cats <- as.vector(unique(Category))
for(cat in cats)
{
corpus <- read.csv(paste(directory,"/",cat,".csv", sep=""))
if(nrow(corpus) > 2)
{
fit <- cmdscale(corpus, k=2)
data_cat <- as.data.frame(fit)
data_cat$Category <- gsub("\\d+ ", "", rownames(data_cat))
data_cat$URLID <- gsub("(\\d+)?.*", "\\1", rownames(fit))
p <- p + geom_point(data=subset(data_cat, URLID != ""),
aes(x=V1, y=V2, size=1.5, color=Category)) +
geom_point(data=subset(data_cat, URLID == ""),
aes(x=V1, y=V2, size=7, shape=c(1)))
}
}
p <- p + scale_x_continuous("", breaks=NA) +
scale_y_continuous("", breaks=NA) +
opts(axis.text.x = theme_blank(), axis.title.x=theme_blank(),
axis.text.y = theme_blank(), axis.title.y=theme_blank(),
legend.position = "none") +
facet_wrap(~ Category)
p
dev.off()