abd-clam 0.11.0

Clustered Learning of Approximate Manifolds
Documentation 
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

import paths
import plots
import reports

data_name = "glove-25"

# Plot lfd-vs-depth
trees_path = paths.REPORTS_DIR.joinpath(
    "mac_2",
    "trees",
)
output_base = trees_path.joinpath("plots")
output_base.mkdir(exist_ok=True)

for path in trees_path.iterdir():
    if "__" not in path.name:
        continue
    if data_name not in path.name:
        continue

    print(f"reading from {path.name} ...")
    tree: reports.TreeReport = reports.TreeReport.parse_file(path.joinpath("tree.json"))
    clusters = reports.load_tree(path)
    clusters_by_depth = [
        [c for c in clusters if len(c.name) == d + 1] for d in range(tree.max_depth)
    ]
    plots.plot_lfd_vs_depth(
        "violin",
        tree,
        clusters_by_depth,
        False,
        output_base,
    )

# Plot search timings
search_path = paths.REPORTS_DIR.joinpath(
    "mac_2",
)
output_base = search_path.joinpath("plots")
output_base.mkdir(exist_ok=True)

for path in search_path.iterdir():
    if not path.name.endswith("json"):
        continue
    if data_name not in path.name:
        continue

    r: reports.RnnReport = reports.RnnReport.parse_file(path)
    assert len(r.is_valid()) == 0
    r.plot(False, output_base)