boltzmann9 0.1.4__py3-none-any.whl → 0.1.7__py3-none-any.whl

boltzmann9/visualization.py

@@ -0,0 +1,115 @@
+"""Visualization utilities for data exploration."""
+
+from __future__ import annotations
+
+from typing import Optional
+
+import numpy as np
+import pandas as pd
+
+
+def plot_simulation_data(
+    df: pd.DataFrame,
+    *,
+    n_bits: int,
+    k_bins: int,
+    start: int = 0,
+    stop: int = 100,
+    figsize: tuple[int, int] = (15, 10),
+) -> None:
+    """Plot simulation data visualization.
+
+    Creates a 6-panel figure showing:
+    1. Continuous vs discretized R values
+    2. Distribution histogram
+    3. Bin indices over time
+    4. Binary representation heatmap
+    5. Bin index distribution
+    6. Decision variable by bin
+
+    Args:
+        df: DataFrame with simulation data.
+        n_bits: Number of bits used for binary encoding.
+        k_bins: Number of discretization bins.
+        start: Start index for time series plots.
+        stop: End index for time series plots.
+        figsize: Figure size tuple.
+    """
+    try:
+        import matplotlib.pyplot as plt
+    except ImportError as e:
+        print(f"matplotlib not available: {e}")
+        return
+
+    fig = plt.figure(figsize=figsize)
+
+    # Plot 1: Continuous vs Discretized
+    ax1 = plt.subplot(3, 2, 1)
+    plt.plot(
+        df["t"][start:stop],
+        df["R_continuous"][start:stop],
+        alpha=0.5,
+        label="Continuous",
+        linewidth=1,
+    )
+    plt.xlabel("Time")
+    plt.ylabel("R")
+    plt.legend()
+    plt.title("Continuous vs Discretized R")
+    plt.grid(True, alpha=0.3)
+
+    # Plot 2: Histogram
+    ax2 = plt.subplot(3, 2, 2)
+    plt.hist(df["R"], bins=k_bins, edgecolor="black", alpha=0.7)
+    plt.xlabel("R (discretized)")
+    plt.ylabel("Frequency")
+    plt.title(f"Distribution across {k_bins} bins")
+    plt.grid(True, alpha=0.3)
+
+    # Plot 3: Bin indices over time
+    ax3 = plt.subplot(3, 2, 3)
+    plt.plot(
+        df["t"][start:stop], df["R_bin_index"][start:stop], drawstyle="steps-post"
+    )
+    plt.xlabel("Time")
+    plt.ylabel("Bin Index")
+    plt.title("Bin Index over Time")
+    plt.yticks(range(k_bins))
+    plt.grid(True, alpha=0.3)
+
+    # Plot 4: Binary bits over time (heatmap style)
+    ax4 = plt.subplot(3, 2, 4)
+    binary_matrix = np.array(
+        [df[f"R_bit_{i}"][start:stop].values for i in range(n_bits)]
+    )
+    plt.imshow(binary_matrix, aspect="auto", cmap="binary", interpolation="nearest")
+    plt.xlabel("Time")
+    plt.ylabel("Bit Position")
+    plt.title(f"Binary Representation over Time ({n_bits} bits)")
+    plt.yticks(range(n_bits), [f"Bit {i}" for i in range(n_bits)])
+    plt.colorbar(label="Bit Value")
+
+    # Plot 5: Bin index distribution
+    ax5 = plt.subplot(3, 2, 5)
+    bin_counts = df["R_bin_index"].value_counts().sort_index()
+    plt.bar(bin_counts.index, bin_counts.values, alpha=0.7, edgecolor="black")
+    plt.xlabel("Bin Index")
+    plt.ylabel("Frequency")
+    plt.title("Bin Index Distribution")
+    plt.xticks(range(k_bins))
+    plt.grid(True, alpha=0.3)
+
+    # Plot 6: Decision variable vs bin index
+    ax6 = plt.subplot(3, 2, 6)
+    df_valid = df[df["x"].notna()]
+    decision_by_bin = df_valid.groupby("R_bin_index")["x"].mean()
+    plt.bar(decision_by_bin.index, decision_by_bin.values, alpha=0.7, edgecolor="black")
+    plt.xlabel("Bin Index")
+    plt.ylabel("Mean Decision (x)")
+    plt.title("Average Decision by Bin")
+    plt.xticks(range(k_bins))
+    plt.ylim([0, 1])
+    plt.grid(True, alpha=0.3)
+
+    plt.tight_layout()
+    plt.show()

{boltzmann9-0.1.4.dist-info → boltzmann9-0.1.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: boltzmann9
-Version: 0.1.4
+Version: 0.1.7
 Summary: Restricted Boltzmann Machine implementation in PyTorch
 License: MIT
 Requires-Python: >=3.10

boltzmann9-0.1.7.dist-info/RECORD

@@ -0,0 +1,19 @@
+boltzmann9/__init__.py,sha256=ziEu9BCO40bLdp0HWpLTD32iwpNQSEBsfltOlz0R87A,960
+boltzmann9/__main__.py,sha256=yj4HraCGopPtcUNTFtgTJ9VaXpVHzUWsnqfMOG04Sas,71
+boltzmann9/cli.py,sha256=mPD6QonSOZ3P3Vc5hznK3Ib3Lf_3ROvLBmcY385BFho,11388
+boltzmann9/config.py,sha256=tNvK6HsToX21Gu3KPCgt_6rOi9yYDpRG53zkniSZ2Wc,1604
+boltzmann9/data.py,sha256=xgaHu6DPaanHrptl8ECq31t6E5yD9MatdVb8JHG-l0U,4011
+boltzmann9/data_generator.py,sha256=aXm8b6O4qoXpDa8TCOdDxF_lDaMtcLNR0P06u2cXb1g,7550
+boltzmann9/model.py,sha256=npBDd_pow4f9gcsLIpWMmJLBB6sGljtaOluvXcFlLP0,31538
+boltzmann9/pipeline.py,sha256=C3L7tHVakrmvdnW9KYMXhXvscwG7TP8xmVHbunx1E8w,5965
+boltzmann9/preprocessor.py,sha256=USxpMCnC80wp4zO72lPxmFeVdfB4I5NE4P2YlojCVPQ,21758
+boltzmann9/project.py,sha256=Iap-nxhhAZDw2Y4yQg53gFQ0CQPCk2zfmZZC0bTPmxQ,5333
+boltzmann9/run_utils.py,sha256=1eRDEkN8j1mG5g9iXArWRA1yQq6pw2agVUGV4Yhy0jk,7823
+boltzmann9/tester.py,sha256=yMYhYKWWiG-LhJQvKv7QmMCYPQLbmtMxFVrJ9iHZVs8,5659
+boltzmann9/utils.py,sha256=8ftNcbV4dz52-t0Ewu4E7CmkkVzPDa6kd8ZnxaBvhMM,1337
+boltzmann9/visualization.py,sha256=iPSytX7GMfmbRbR43xOmybLZBLmy58D4Rp8wasISMcs,3407
+boltzmann9-0.1.7.dist-info/METADATA,sha256=4i9cMFOYamgkAeYOCgNl-2P000hYO_FLa38nPihsSNU,3097
+boltzmann9-0.1.7.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+boltzmann9-0.1.7.dist-info/entry_points.txt,sha256=Y2wY1lqICCjTxuaPgme2QxkoAknrASKmhLljcb5FEyw,51
+boltzmann9-0.1.7.dist-info/top_level.txt,sha256=AE0_urGKOFSgKxXf7WQfjGDROWYppyI4tw1VIg-paQI,11
+boltzmann9-0.1.7.dist-info/RECORD,,

boltzmann9-0.1.7.dist-info/top_level.txt

@@ -0,0 +1 @@
+boltzmann9

boltzmann9-0.1.4.dist-info/RECORD

@@ -1,5 +0,0 @@
-boltzmann9-0.1.4.dist-info/METADATA,sha256=6uk2GRe3SD0UDl1Q69PtluVsIr9UjRG4Y9OWIc2swFs,3097
-boltzmann9-0.1.4.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-boltzmann9-0.1.4.dist-info/entry_points.txt,sha256=Y2wY1lqICCjTxuaPgme2QxkoAknrASKmhLljcb5FEyw,51
-boltzmann9-0.1.4.dist-info/top_level.txt,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-boltzmann9-0.1.4.dist-info/RECORD,,

boltzmann9-0.1.4.dist-info/top_level.txt

@@ -1 +0,0 @@
-