Skip to content

Commit 93d22da

Browse files
Sem print methods
1 parent be33c72 commit 93d22da

1 file changed

Lines changed: 14 additions & 13 deletions

File tree

  • src/frontend/specification

src/frontend/specification/Sem.jl

Lines changed: 14 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -28,20 +28,17 @@ for f in (
2828
end
2929

3030
function Base.show(io::IO, term::LossTerm)
31+
print(io, nameof(losstype(term)))
32+
print(io, "\n")
3133
if !isnothing(id(term))
32-
print(io, ":$(id(term)): ")
34+
print(io, " - id: $(id(term)) \n")
3335
end
34-
print(io, nameof(losstype(term)))
3536
if issemloss(term)
36-
print(
37-
io,
38-
" ($(nsamples(term)) samples, $(nobserved_vars(term)) observed, $(nlatent_vars(term)) latent variables)",
39-
)
37+
print(io, " - observed: $(nameof(typeof(observed(loss(term))))) \n")
38+
print(io, " - implied: $(nameof(typeof(implied(loss(term))))) \n")
4039
end
4140
if !isnothing(weight(term))
42-
print(io, " w=$(round(weight(term), digits=3))")
43-
else
44-
print(io, " w=1")
41+
print(io, " - weight: $(round(weight(term), digits=3))")
4542
end
4643
end
4744

@@ -604,12 +601,16 @@ end
604601
# pretty printing
605602
##############################################################
606603

604+
_subtype_info(::Sem) = ""
605+
_subtype_info(::SemFiniteDiff) = " : Finite Difference Approximation"
606+
607607
function Base.show(io::IO, sem::AbstractSem)
608-
println(io, "Structural Equation Model ($(nameof(typeof(sem))))")
609-
println(io, "- $(nparams(sem)) parameters")
610-
println(io, "- Loss terms:")
608+
print(io, "Structural Equation Model")
609+
print(io, _subtype_info(sem))
610+
print("\n")
611+
print(io, "- Loss Functions \n")
611612
for term in loss_terms(sem)
612-
print(io, " - ")
613+
print(io, " > ")
613614
print(io, term)
614615
println(io)
615616
end

0 commit comments

Comments
 (0)