tests/message-passing_tests/schedules_and_rules_test1.R

In npetraco/CRFutil: Handy Utility Functions for Use with CRF and gRbase packages

library(CRFutil)

# Model:
grphf <- ~1:3 + 2:3 + 3:4 + 3:5 # Node names must be numbers for now.... FIX
plot(ug(grphf))
dev.off()

# Convert graph to to pair-wise factor graph
pwfg <- mrf2pwfg(grphf, plotQ=T)

# It'd be nice to have a neighbor function with ability to drop some neighbors.
# We need this when using makef,v
# We can get neighbors this way using graph
nodes(pwfg)
adj(pwfg, "1")
adj(pwfg, "f1")
adj(pwfg, "f12")

# OR we can do it this way with igraph. Need to convert graphNEL to igraph format first
igpwfg <- graph_from_graphnel(pwfg)
V(igpwfg)
neighborhood(igpwfg, nodes = V(igpwfg)[4], mindist = 1)


# Make mailboxes for each node
fg.msgs <- rep(list(NULL), length(fg.node.nms))
for(i in 1:length(fg.node.nms)){
  nei.nms   <- as.vector(unlist(adj(pwfg, fg.node.nms[i]))) # Names of neighboring nodes to node i
  #print(nei.nms)

  msg.box <-  rep(list(NULL), length(nei.nms))
  print(fg.node.nms[i])
  #print(nei.nms)
  #print(msg.box)
  names(msg.box) <- nei.nms
  print(msg.box)
  print("--------")

  fg.msgs[[i]] <- msg.box

}
names(fg.msgs) <- fg.node.nms

# Accessing the required mailbox....
names(fg.msgs)
fg.msgs[["f12"]]
fg.msgs[[fg.node.nms[3]]]



# Message passing tests:
# Schedules:
schs <- get.root.paths(pwfg, root.node = 2, serial.schedsQ = T)
schs$forwrd
schs$backward

schs$forwrd
schs$forwrd[,c(1,2)]

for(j in 1:(length(schs$forwrd[1,])-1)) {

  print(paste0(schs$forwrd[1,j], "-->",schs$forwrd[1,j+1]))

  # Check for NA on right node. Means end of that chain
  st.nme  <- schs$forwrd[1,j] # left (starting) node name
  sp.nme  <- schs$forwrd[1,j+1] # right (ending) node name

  # Pass rules:
  f.nodeQ <- "f" %in% unlist(strsplit(st.nme,split = ""))

  if(f.nodeQ == T) {
    # Y_k \in \text{ne}(\Psi_{\text{o}})\backslash X
    print(paste0("Neighbors of ", st.nme, ":"))
    ne.f <- adj(pwfg, st.nme)[[1]]
    #ne.f <- ne.f[-which(ne.f == sp.nme)]
    print(ne.f)
  }

}




# Message: f1 -> 1 # HOW TO BETTER NAME OR STORE MESSAGES??????
np[[1]]
m.f1.1 <- make.f2v.msg(in.v.msgs.list = NULL, f.msg = np[[1]], out.v.nme = 1)
m.f1.1

# Message: f2 -> 2
np[[2]]
m.f2.2 <- make.f2v.msg(in.v.msgs.list = NULL, f.msg = np[[2]], out.v.nme = 2)
m.f2.2

#--------------------------------------------------------
schs$forwrd[,c(2,3)]
# Message: 1 -> f12
m.1.f12 <- make.v2f.msg(list(m.f1.1))
m.1.f12
make.v2f.msg(m.f1.1)

# Message: f12 -> 2


make.f2v.msg(
  in.v.msgs.list= lapply(c(1,3), function(xx){ep[[xx]]}),
  f.msg = ep[[4]],
  out.v.nme = 1)

schs$forwrd
m.f1.1 <- make.f2v.msg(f.msg = np[[1]], out.v.nme = 1)
m.f2.2 <- make.f2v.msg(f.msg = np[[2]], out.v.nme = 2)
m.f3.3 <- make.f2v.msg(f.msg = np[[3]], out.v.nme = 3)
m.f4.4 <- make.f2v.msg(f.msg = np[[4]], out.v.nme = 4)
m.f5.5 <- make.f2v.msg(f.msg = np[[5]], out.v.nme = 5)

m.1.f15 <- make.v2f.msg(list(m.f1.1))
m.2.f12 <- make.v2f.msg(list(m.f2.2))
m.3.f13 <- make.v2f.msg(list(m.f3.3))
m.4.f14 <- make.v2f.msg(list(m.f4.4))

m.f15.5 <- make.f2v.msg(in.v.msgs.list = list(m.1.f15), f.msg = ep[[4]], out.v.nme = 5)