Engine prototype #13
@ -1,105 +0,0 @@
|
||||
include("Engine.jl")
|
||||
|
||||
using SparseArrays
|
||||
|
||||
# --- construct the nucleus spheres ---
|
||||
|
||||
println("--- Nucleus spheres ---\n")
|
||||
|
||||
# initialize the partial gram matrix for the circumscribing and nucleus spheres
|
||||
J = Int64[]
|
||||
K = Int64[]
|
||||
values = BigFloat[]
|
||||
for n in 1:3
|
||||
push!(J, n)
|
||||
push!(K, n)
|
||||
push!(values, 1)
|
||||
if n > 1
|
||||
append!(J, [1, n])
|
||||
append!(K, [n, 1])
|
||||
append!(values, [1, 1])
|
||||
end
|
||||
end
|
||||
gram_nuc = sparse(J, K, values)
|
||||
|
||||
# make an initial guess
|
||||
guess_nuc = hcat(
|
||||
Engine.sphere(BigFloat[0, 0, 0], BigFloat(15)),
|
||||
Engine.sphere(BigFloat[0, 0, -10], BigFloat(5)),
|
||||
Engine.sphere(BigFloat[0, 0, 11], BigFloat(3)),
|
||||
)
|
||||
frozen_nuc = [CartesianIndex(4, k) for k in 1:3]
|
||||
|
||||
# complete the gram matrix using Newton's method with backtracking
|
||||
L_nuc, success_nuc, history_nuc = Engine.realize_gram(gram_nuc, guess_nuc, frozen_nuc)
|
||||
completed_gram_nuc = L_nuc'*Engine.Q*L_nuc
|
||||
println("Completed Gram matrix:\n")
|
||||
display(completed_gram_nuc)
|
||||
if success_nuc
|
||||
println("\nTarget accuracy achieved!")
|
||||
else
|
||||
println("\nFailed to reach target accuracy")
|
||||
end
|
||||
println("Steps: ", size(history_nuc.scaled_loss, 1))
|
||||
println("Loss: ", history_nuc.scaled_loss[end], "\n")
|
||||
|
||||
# --- construct the chain of spheres ---
|
||||
|
||||
# initialize the partial gram matrix for the chain of spheres
|
||||
J = Int64[]
|
||||
K = Int64[]
|
||||
values = BigFloat[]
|
||||
for a in 4:9
|
||||
push!(J, a)
|
||||
push!(K, a)
|
||||
push!(values, 1)
|
||||
|
||||
# each chain sphere is internally tangent to the circumscribing sphere
|
||||
append!(J, [a, 1])
|
||||
append!(K, [1, a])
|
||||
append!(values, [1, 1])
|
||||
|
||||
# each chain sphere is externally tangent to the nucleus spheres
|
||||
for n in 2:3
|
||||
append!(J, [a, n])
|
||||
append!(K, [n, a])
|
||||
append!(values, [-1, -1])
|
||||
end
|
||||
|
||||
# each chain sphere is externally tangent to the next sphere in the chain
|
||||
#=
|
||||
a_next = 4 + mod(a-3, 6)
|
||||
append!(J, [a, a_next])
|
||||
append!(K, [a_next, a])
|
||||
append!(values, [-1, -1])
|
||||
=#
|
||||
end
|
||||
gram_chain = sparse(J, K, values)
|
||||
|
||||
if success_nuc
|
||||
println("--- Chain spheres ---\n")
|
||||
|
||||
# make an initial guess, with the circumscribing and nucleus spheres included
|
||||
# as frozen elements
|
||||
guess_chain = hcat(
|
||||
L_nuc,
|
||||
(
|
||||
Engine.sphere(10*BigFloat[cos(k*π/3), sin(k*π/3), 0], BigFloat(2.5))
|
||||
for k in 1:6
|
||||
)...
|
||||
)
|
||||
frozen_chain = [CartesianIndex(j, k) for k in 1:3 for j in 1:5]
|
||||
|
||||
# complete the gram matrix using Newton's method with backtracking
|
||||
L_chain, success_chain, history_chain = Engine.realize_gram(gram_chain, guess_chain, frozen_chain)
|
||||
completed_gram_chain = L_chain'*Engine.Q*L_chain
|
||||
println("Completed Gram matrix:\n")
|
||||
display(completed_gram_chain)
|
||||
if success_chain
|
||||
println("\nTarget accuracy achieved!")
|
||||
else
|
||||
println("\nFailed to reach target accuracy")
|
||||
end
|
||||
println("Steps: ", size(history_chain.scaled_loss, 1))
|
||||
println("Loss: ", history_chain.scaled_loss[end], "\n")
|
||||
end
|
Loading…
Reference in New Issue
Block a user