Rel 0.1.0

Author: goedman

src/methods/induced_covariance_graph.jl

@@ -10,7 +10,7 @@ function ancester_graph(e::NamedArray)
   if sum(size(e)) == 0
     return(e)
   end
-  indicator_matrix(Int.(inv(2 * I(size(e, 1)) - e)))
+  indicator_matrix(Int.(round.(Int, inv(2 * I(size(e, 1)) - e))))
 end
 
 """

test/test_descendents_01.dot

@@ -0,0 +1,14 @@
+digraph test_desc {
+  v -> y;
+  v -> x;
+  y -> k;
+  y -> w;
+  k -> s2;
+  z -> x;
+  x -> u;
+  w -> u;
+  w -> s2;
+  u -> s1;
+  s1 -> s3;
+  s3 -> s2;
+}

test/test_descendents_01.jl

@@ -1,29 +1,27 @@
 using StructuralCausalModels
 
+ProjDir = @__DIR__
+
 d_str = "DAG(u ~ x + w, x ~ v + z, y ~ v, s1 ~ u, w ~ y, k ~ y, s2 ~ w + s3 + k, s3 ~ s1)"
 
 dag = DAG("test_desc", d_str)
 
 to_dagitty(dag) |> display
 
+#=
 fname = joinpath(ProjDir, "test_descendents_01.dot")
 to_graphviz(dag, fname)
 Sys.isapple() && run(`open -a GraphViz.app $(fname)`)
+=#
 
 display(dag)
 display(dag.a)
 println(dag.a[:x, :u])
 println(dag.e[:x, :u])
 println()
 
-@show d_separation(dag, :x, :w)
-@show d_separation(dag, :x, :w; cset=:v)
-@show d_separation(dag, :x, :w; cset=:y)
-@show d_separation(dag, :x, :v; cset=:s3)
-println()
+bs = basis_set(dag)
+display(bs)
 
-@show adjustment_sets(dag, :x, :w)
-@show adjustment_sets(dag, :x, :s1)
-@show adjustment_sets(dag, :x, :s2)
-@show adjustment_sets(dag, :u, :k)
-@show adjustment_sets(dag, :y, :s2)
+as = adjustment_sets(dag, :u, :s2)
+display(as)

test/test_descendents_02.jl

@@ -8,6 +8,10 @@ dagitty_str = to_dagitty(dag)
 
 fname = joinpath(ProjDir, "test_descendents_02.dot")
 to_graphviz(dag, fname)
-#Sys.isapple() && run(`open -a GraphViz.app $(fname)`)
+Sys.isapple() && run(`open -a GraphViz.app $(fname)`)
 
 bs = basis_set(dag)
+display(bs)
+
+as = adjustment_sets(dag, :u, :s2)
+display(as)

test/test_descendents_03.jl

@@ -13,77 +13,6 @@ Sys.isapple() && run(`open -a GraphViz.app $(fname)`)
 bs = basis_set(dag)
 bs |> display
 
-bs[9] |> display
-@show d_separation(dag, :x1, :x2; cset=[:v])
-println()
+as = adjustment_sets(dag, :x2, :y)
+display(as)
 
-bs[10] |> display
-@show d_separation(dag, :x1, :x3)
-println()
-
-bs[11] |> display
-@show d_separation(dag, :x1, :k1; cset=[:w])
-println()
-
-bs[12] |> display
-@show d_separation(dag, :x1, :k2; cset=[:k2])
-println()
-
-bs[13] |> display
-@show d_separation(dag, :x1, :y; cset=[:x2, :x3, :k2])
-println()
-
-bs[1] |> display
-@show d_separation(dag, :k1, :v, cset=[:x1])
-@show d_separation(dag, :k1, :v, cset=[:w])
-@show d_separation(dag, :k1, :x1, cset=[:w])
-@show d_separation(dag, :k1, :x2, cset=[:v])
-@show d_separation(dag, :k1, :x2, cset=[:x1])
-@show d_separation(dag, :k1, :x2, cset=[:w])
-@show d_separation(dag, :k1, :x3, cset=[:w])
-@show d_separation(dag, :k1, :y, cset=[:k2, :x2, :x3])
-@show d_separation(dag, :k1, :y, cset=[:k2, :v, :x3])
-@show d_separation(dag, :k1, :y, cset=[:k2, :x1, :x3])
-@show d_separation(dag, :k1, :y, cset=[:k2, :w])
-println()
-
-@show d_separation(dag, :k2, :v, cset=[:x1])
-@show d_separation(dag, :k2, :v, cset=[:w])
-@show d_separation(dag, :k2, :v, cset=[:k1])
-@show d_separation(dag, :k2, :w, cset=[:k1])
-@show d_separation(dag, :k2, :x1, cset=[:w])
-@show d_separation(dag, :k2, :x1, cset=[:k1])
-@show d_separation(dag, :k2, :x2, cset=[:v])
-@show d_separation(dag, :k2, :x2, cset=[:x1])
-@show d_separation(dag, :k2, :x2, cset=[:w])
-@show d_separation(dag, :k2, :x2, cset=[:k1])
-@show d_separation(dag, :k2, :x3, cset=[:w])
-@show d_separation(dag, :k2, :x3, cset=[:k1])
-println()
-
-@show d_separation(dag, :v, :w; cset=:x1)
-@show d_separation(dag, :v, :x3)
-@show d_separation(dag, :v, :y; cset=[:k2, :x2, :x3])
-@show d_separation(dag, :v, :y; cset=[:k1, :x2, :x3])
-@show d_separation(dag, :v, :y; cset=[:w, :x2, :x3])
-@show d_separation(dag, :v, :y; cset=[:x2, :x1])
-println()
-
-@show d_separation(dag, :w, :x2; cset=[:v])
-@show d_separation(dag, :w, :x2; cset=[:x1])
-@show d_separation(dag, :w, :y; cset=[:k2, :x2, :x3])
-@show d_separation(dag, :w, :y; cset=[:k2, :v, :x3])
-@show d_separation(dag, :w, :y; cset=[:k1, :x2, :x3])
-@show d_separation(dag, :w, :y; cset=[:k1, :v, :x3])
-@show d_separation(dag, :w, :y; cset=[:k1, :x1, :x3])
-@show d_separation(dag, :w, :y; cset=[:k2, :x2, :x3])
-println()
-
-@show d_separation(dag, :x1, :y; cset=[:k1, :v, :x3])
-@show d_separation(dag, :x1, :y; cset=[:v, :w, :x3])
-println()
-
-@show d_separation(dag, :x2, :x3)
-println()
-
-@show adjustment_sets(dag, :x2, :y)

test/test_open_paths_04.jl

@@ -23,6 +23,7 @@ to_graphviz(dag, fname)
   :h => :i, :g => :h, :f => :g, :c => :f, :b => :c, :a => :b, 
   :q => :r, :p => :q, :o => :p, :n => :o, :m => :n, :k => :l, 
   :l => :m, :i => :k)
+
 @test length(bs) == 272
 @test bs[3][1] == :D
 @test bs[3][2] == :a