likelihood 1.2.20__py3-none-any.whl → 1.2.21__py3-none-any.whl

likelihood/main.py

@@ -3,10 +3,9 @@ from typing import Callable, List, Tuple
 import corner
 import matplotlib.pyplot as plt
 import numpy as np
-from numpy import ndarray
 
 
-def lnprior(theta: ndarray, conditions: List[Tuple[float, float]]) -> float:
+def lnprior(theta: np.ndarray, conditions: List[Tuple[float, float]]) -> float:
     """Computes the prior probability.
 
     Parameters
@@ -22,29 +21,22 @@ def lnprior(theta: ndarray, conditions: List[Tuple[float, float]]) -> float:
     lp : `float`
         The a priori probability.
     """
+    if len(conditions) != 2 * len(theta):
+        raise ValueError("Length of conditions must be twice the length of theta.")
 
-    try:
-        if len(conditions) != 2 * len(theta):
-            error_type = "IndexError"
-            msg = "Length of conditions must be twice the length of theta."
-            print(f"{error_type}: {msg}")
-        else:
-            cond = np.array(conditions).reshape((len(theta), 2))
-            for i in range(len(theta)):
-                if cond[i, 0] < theta[i] < cond[i, 1]:
-                    lp = 0.0
-                else:
-                    return np.inf
-            return lp
-    except:
-        return 0.0
+    cond = np.array(conditions).reshape((len(theta), 2))
+    within_bounds = np.logical_and(cond[:, 0] < theta, theta < cond[:, 1])
+    if not np.all(within_bounds):
+        return np.inf
+
+    return 0.0
 
 
 def fun_like(
-    x: ndarray,
-    y: ndarray,
+    x: np.ndarray,
+    y: np.ndarray,
     model: Callable,
-    theta: ndarray,
+    theta: np.ndarray,
     conditions: List[Tuple[float, float]] = None,
     var2: float = 1.0,
 ) -> float:
@@ -57,14 +49,14 @@ def fun_like(
     y : `np.ndarray`
         An $n$ dimensional array that will be compared with model's output.
     model : `Callable`
-        A Python function defined by the user. This function should recieve
-        two arguments $(`x`, `theta`)$.
+        A Python function defined by the user. This function should receive
+        two arguments $(x, theta)$.
     theta : `np.ndarray`
         The array containing the model's parameters.
-    conditions : `list`
+    conditions : `list`, optional
         A list containing $2n$-conditions for the (min, max) range of the
-        $n$ parameters.
-    var2 : `float`
+        $n$ parameters. Defaults to None.
+    var2 : `float`, optional
         Determines the step size of the walker. By default it is set to `1.0`.
 
     Returns
@@ -72,28 +64,25 @@ def fun_like(
     lhood : `float`
         The computed likelihood.
     """
-
-    lp = lnprior(theta, conditions)
-    inv_sigma2 = 1.0 / (var2)
+    lp = 0.0 if conditions is None else lnprior(theta, conditions)
+    inv_sigma2 = 1.0 / var2
     y_hat = model(x, theta)
 
     try:
         y_hat.shape[1]
-    except:
+    except IndexError:
         y_hat = y_hat[np.newaxis, ...].T
 
     y_sum = np.sum((y - y_hat) ** 2 * inv_sigma2 - np.log(inv_sigma2))
     lhood = 0.5 * y_sum
 
-    if not np.isfinite(lp):
-        lhood = np.inf
-    else:
-        lhood += lp
+    if not np.isfinite(lhood):
+        return np.inf
 
-    return lhood
+    return lhood + lp
 
 
-def update_theta(theta: ndarray, d: float) -> ndarray:
+def update_theta(theta: np.ndarray, d: float) -> np.ndarray:
     """Updates the theta parameters.
 
     Parameters
@@ -108,21 +97,18 @@ def update_theta(theta: ndarray, d: float) -> ndarray:
     theta_new : `np.array`
         An ndarray with the updated theta values.
     """
-
-    theta_new = [np.random.normal(theta[k], d / 2.0) for k in range(len(theta))]
-
-    return theta_new
+    return np.random.normal(theta, d / 2.0)
 
 
 def walk(
-    x: ndarray,
-    y: ndarray,
+    x: np.ndarray,
+    y: np.ndarray,
     model: Callable,
-    theta: ndarray,
+    theta: np.ndarray,
     conditions: List[Tuple[float, float]] = None,
     var2: float = 0.01,
     mov: int = 100,
-    d: int = 1,
+    d: float = 1.0,
     tol: float = 1e-4,
     mode: bool = True,
 ):
@@ -135,25 +121,25 @@ def walk(
     y : np.ndarray
         An $n$ dimensional array that will be compared with model's output.
     model : `Callable`
-        A Python function defined by the user. This function should recieve
+        A Python function defined by the user. This function should receive
         two arguments $(x, theta)$.
     theta : `np.ndarray`
         The array containing the model's parameters.
-    conditions : `list`
+    conditions : `list`, optional
         A list containing $2n$-conditions for the (min, max) range of the
-        $n$ parameters.
-    var2 : `float`
+        $n$ parameters. Defaults to None.
+    var2 : `float`, optional
         Determines the step size of the walker. By default it is set to `1.0`.
-    mov : `int`
+    mov : `int`, optional
         Number of movements that walker will perform. By default it is set
         to `100`.
-    d : `float`
+    d : `float`, optional
         Size of the Gaussian step for the walker.
-    tol : `float`
+    tol : `float`, optional
         Convergence criteria for the log-likelihood. By default it is set
         to `1e-3`.
-    mode : `bool`
-        By default it is set to `True`.
+    mode : `bool`, optional
+        Defaults to `True`.
 
     Returns
     -------
@@ -164,43 +150,29 @@ def walk(
     y0 : `float`
         The log-likelihood value.
     """
-
-    greach = False
     nwalk = []
 
     for i in range(mov):
         nwalk.append(theta)
         theta_new = update_theta(theta, d)
 
-        if not greach:
-            y0 = fun_like(x, y, model, theta, conditions, var2)
-            y1 = fun_like(x, y, model, theta_new, conditions, var2)
-
-            if y0 <= tol and mode:
+        y0 = fun_like(x, y, model, theta, conditions, var2)
+        y1 = fun_like(x, y, model, theta_new, conditions, var2)
+        if y0 <= tol or y1 <= tol:
+            if mode:
                 print("Goal reached!")
-                greach = True
-
-                return theta, nwalk, y0
-            else:
-                if y1 <= tol and mode:
-                    print("Goal reached!")
-                    greach = True
-
-                    return theta_new, nwalk, y1
-                else:
-                    ratio = y0 / y1
-                    boltz = np.random.rand(1)
-                    prob = np.exp(-ratio)
-
-                    if y1 < y0:
-                        theta = theta_new
-                        theta_new = update_theta(theta, d)
-                    else:
-                        if prob > boltz:
-                            theta = theta_new
-                            theta_new = update_theta(theta, d)
-                        else:
-                            theta_new = update_theta(theta, d)
+            return (theta_new, nwalk, y1) if y1 <= tol else (theta, nwalk, y0)
+
+        if y1 >= y0:
+            ratio = y0 / y1
+            prob = np.exp(-ratio)
+
+            if prob > np.random.rand():
+                theta = theta_new
+        else:
+            theta = theta_new
+            theta_new = update_theta(theta, d)
+
     if mode:
         print("Maximum number of iterations reached!")
         print(f"The log-likelihood is: {y0}")
@@ -210,14 +182,14 @@ def walk(
 
 def walkers(
     nwalkers: int,
-    x: ndarray,
-    y: ndarray,
+    x: np.ndarray,
+    y: np.ndarray,
     model: Callable,
-    theta: ndarray,
-    conditions: bool = None,
+    theta: np.ndarray,
+    conditions: List[Tuple[float, float]] = None,
     var2: float = 0.01,
     mov: int = 100,
-    d: int = 1,
+    d: float = 1.0,
     tol: float = 1e-4,
     mode: bool = False,
     figname: str = "fig_out.png",
@@ -233,27 +205,27 @@ def walkers(
     y : `np.ndarray`
         An $n$ dimensional array that will be compared with model's output.
     model : `Callable`
-        A Python function defined by the user. This function should recieve
+        A Python function defined by the user. This function should receive
         two arguments $(x, theta)$.
     theta : `np.ndarray`
         The array containing the model's parameters.
-    conditions : `list`
+    conditions : `list`, optional
         A list containing $2n$-conditions for the (min, max) range of the
-        $n$ parameters.
-    var2 : `float`
+        $n$ parameters. Defaults to None.
+    var2 : `float`, optional
         Determines the step size of the walker. By default it is set to `1.0`.
-    mov : `int`
+    mov : `int`, optional
         Number of movements that walker will perform. By default it is set
         to `100`.
-    d : `float`
+    d : `float`, optional
         Size of the Gaussian step for the walker.
-    tol : `float`
-        Convergence criteria for the log-likelihhod. By default it is set
+    tol : `float`, optional
+        Convergence criteria for the log-likelihood. By default it is set
         to `1e-3`.
-    mode : `bool`
+    mode : `bool`, optional
         Specifies that we will be working with more than one walker. By
         default it is set to `False`.
-    figname : `str`
+    figname : `str`, optional
         The name of the output file for the figure. By default it is set
         to `fig_out.png`.
 
@@ -264,7 +236,6 @@ def walkers(
     error : `np.array`
         The log-likelihood array.
     """
-
     error = []
     par = []
 
@@ -274,30 +245,29 @@ def walkers(
         nwalk = np.array(nwalk).reshape((len(nwalk), len(nwalk[i])))
         error.append(y0)
 
-        if figname != None:
+        if figname:
             for k in range(nwalk.shape[1]):
-                sub = "$\\theta _{" + str(k) + "}$"
+                sub = f"$\\theta _{k}$"
                 plt.plot(range(len(nwalk[:, k])), nwalk[:, k], "-", label=sub)
                 plt.ylabel("$\\theta$")
                 plt.xlabel("iterations")
-                plt.savefig("walkers_" + figname, dpi=300, transparent=True)
+                plt.savefig(f"walkers_{figname}", dpi=300, transparent=True)
 
-    if figname != None:
+    if figname:
         plt.show()
 
-    if nwalk.shape[1] == 2:
-        if figname != None:
-            fig = corner.hist2d(
-                nwalk[:, 0],
-                nwalk[:, 1],
-                range=None,
-                bins=18,
-                smooth=True,
-                plot_datapoints=True,
-                plot_density=True,
-            )
-            plt.ylabel("$\\theta_{1}$")
-            plt.xlabel("$\\theta_{0}$")
-            plt.savefig("theta_" + figname, dpi=300, transparent=True)
+    if len(theta) == 2 and figname:
+        corner.hist2d(
+            nwalk[:, 0],
+            nwalk[:, 1],
+            range=None,
+            bins=18,
+            smooth=True,
+            plot_datapoints=True,
+            plot_density=True,
+        )
+        plt.ylabel("$\\theta_{1}$")
+        plt.xlabel("$\\theta_{0}$")
+        plt.savefig(f"theta_{figname}", dpi=300, transparent=True)
 
     return par, error

likelihood/models/deep/autoencoders.py

@@ -1,3 +1,4 @@
+import logging
 import os
 from functools import partial
 from shutil import rmtree
@@ -6,9 +7,12 @@ import keras_tuner
 import numpy as np
 import pandas as pd
 import tensorflow as tf
-from likelihood.tools import OneHotEncoder
 from pandas.core.frame import DataFrame
 
+from likelihood.tools import OneHotEncoder
+
+logging.getLogger("tensorflow").setLevel(logging.ERROR)
+
 tf.compat.v1.logging.set_verbosity(tf.compat.v1.logging.ERROR)
 
 
@@ -141,7 +145,7 @@ def call_existing_code(
     )
     model.compile(
         optimizer=optimizer,
-        loss="categorical_crossentropy",
+        loss=tf.keras.losses.CategoricalCrossentropy(),
         metrics=[tf.keras.metrics.F1Score(threshold=threshold)],
     )
     return model
@@ -250,9 +254,9 @@ def setup_model(
     if train_mode:
         # Create a new directory if it does not exist
         try:
-            if not os.path.exists(directory):
+            if (not os.path.exists(directory)) and directory != "./":
                 os.makedirs(directory)
-            else:
+            elif directory != "./":
                 print(f"Directory {directory} already exists, it will be deleted.")
                 rmtree(directory)
                 os.makedirs(directory)

likelihood/tools/numeric_tools.py

@@ -8,6 +8,93 @@ from pandas.core.frame import DataFrame
 
 
 # -------------------------------------------------------------------------
+def get_metrics(dataset, actual_column_name, predicted_column_name, verbose=False):
+    # Variables to keep track of the number of correct and total predictions
+    true_positives = 0  # Correctly predicted positives
+    true_negatives = 0  # Correctly predicted negatives
+    false_positives = 0  # Negatives predicted as positives
+    false_negatives = 0  # Positives predicted as negatives
+    total_predictions = len(dataset)
+
+    # Counters for actual and predicted classes
+    actual_positive_count = 0
+    actual_negative_count = 0
+    predicted_positive_count = 0
+    predicted_negative_count = 0
+
+    for index, row in dataset.iterrows():
+        actual_class = row[actual_column_name]
+        predicted_class = row[predicted_column_name]
+
+        # Update confusion matrix counts
+        if actual_class == 1 and predicted_class == 1:  # True positive
+            true_positives += 1
+        elif actual_class == 0 and predicted_class == 0:  # True negative
+            true_negatives += 1
+        elif actual_class == 0 and predicted_class == 1:  # False positive
+            false_positives += 1
+        elif actual_class == 1 and predicted_class == 0:  # False negative
+            false_negatives += 1
+
+        # Update class counts
+        if actual_class == 1:
+            actual_positive_count += 1
+        else:
+            actual_negative_count += 1
+
+        if predicted_class == 1:
+            predicted_positive_count += 1
+        else:
+            predicted_negative_count += 1
+
+    # Calculate accuracy
+    accuracy = (true_positives + true_negatives) / total_predictions * 100
+
+    # Calculate precision
+    if true_positives + false_positives > 0:
+        precision = true_positives / (true_positives + false_positives) * 100
+    else:
+        precision = 0  # Avoid division by zero
+
+    # Calculate recall
+    if true_positives + false_negatives > 0:
+        recall = true_positives / (true_positives + false_negatives) * 100
+    else:
+        recall = 0  # Avoid division by zero
+
+    # Calculate F1-Score
+    if precision + recall > 0:
+        f1_score = 2 * (precision * recall) / (precision + recall)
+    else:
+        f1_score = 0  # Avoid division by zero
+
+    coeff_1 = (true_positives + false_positives) * (false_positives + true_negatives)
+    coeff_2 = (true_positives + false_negatives) * (false_negatives + true_negatives)
+    if coeff_1 + coeff_2 > 0:
+        kappa = (
+            2
+            * (true_positives * true_negatives - false_negatives * false_positives)
+            / (coeff_1 + coeff_2)
+        )
+
+    metrics = {
+        "accuracy": accuracy,
+        "precision": precision,
+        "recall": recall,
+        "f1_score": f1_score,
+        "kappa": kappa,
+    }
+
+    if verbose:
+        print(f"Accuracy: {accuracy:.2f}%")
+        print(f"Precision: {precision:.2f}%")
+        print(f"Recall: {recall:.2f}%")
+        print(f"F1-Score: {f1_score:.2f}")
+        print(f"Cohen's Kappa: {kappa:.4f}")
+    else:
+        return metrics
+
+
 def xi_corr(df: pd.DataFrame) -> pd.DataFrame:
     """Calculate new coefficient of correlation for all pairs of columns in a `DataFrame`.
 

{likelihood-1.2.20.dist-info → likelihood-1.2.21.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: likelihood
-Version: 1.2.20
+Version: 1.2.21
 Summary: A package that performs the maximum likelihood algorithm.
 Home-page: https://github.com/jzsmoreno/likelihood/
 Author: J. A. Moreno-Guerra

{likelihood-1.2.20.dist-info → likelihood-1.2.21.dist-info}/RECORD

@@ -1,5 +1,5 @@
 likelihood/__init__.py,sha256=5C0hapdsk85XZhN_rssRAEFpkRRuKNtj6cyRbqD2_gM,994
-likelihood/main.py,sha256=prqT9egu3B2rcbsVMqYxuosNbe7NhDBCmmZtQ21aSlQ,8591
+likelihood/main.py,sha256=fcCkGOOWKjfvw2tLVqjuKPV8t0rVCIT9FlbYcOv4EYo,7974
 likelihood/graph/__init__.py,sha256=6TuFDfmXTwpLyHl7_KqBfdzW6zqHjGzIFvymjFPlvjI,21
 likelihood/graph/graph.py,sha256=wKJqgxXiSbnvzyW3SjhQVrqp00yKMHf3ph6CIDNVhNM,2891
 likelihood/graph/nn.py,sha256=XqTnAHzXP0jSdLd0IOFjVZUZTcQU-XppsZLmJrG2GMo,12372
@@ -8,12 +8,12 @@ likelihood/models/regression.py,sha256=9cakyGlJCEO6WfpoKLh3GxdXQeQp7cUvJIkQ5odT0
 likelihood/models/simulation.py,sha256=mdgQPg_LEY5svPaF4TFv-DoQRE2oP2ig_uXnwINtewM,4039
 likelihood/models/utils.py,sha256=VtEj07lV-GRoWraQgpfjU0jTt1Ntf9MXgYwe6XYQh20,1552
 likelihood/models/deep/__init__.py,sha256=-KIPippVaMqgG8mEgYjNxYQdqOUcFhUuKhbVe8TTCfo,28
-likelihood/models/deep/autoencoders.py,sha256=s-3ZskcNtB3kg1lQD8KzJ0ff3R_OtVsePTYeShE4FxU,9940
+likelihood/models/deep/autoencoders.py,sha256=2P--nS96XwMi44q0OIxvIp6Mdbt-B4LqwCSXTn2jYrY,10070
 likelihood/tools/__init__.py,sha256=MCjsCWfBNKE2uMN0VizDN1uFzZ_md0X2WZeBdWhrCR8,50
-likelihood/tools/numeric_tools.py,sha256=cPTPgdww2ofxfyhJDomqvtXDgsSDs9iRQ7GHLt5Vl6M,8457
+likelihood/tools/numeric_tools.py,sha256=FA44kbiAcxcquz1el_g3Pqsp5ii8XFkAIrsMs5bGkj0,11445
 likelihood/tools/tools.py,sha256=O39aPxTNsaBVSJFIkNsUESNSkfG4C7GG77wcR51a8IQ,42543
-likelihood-1.2.20.dist-info/LICENSE,sha256=XWHWt9egYEUHGPTnlcZfJKLPmysacOwdiLj_-J7Z9ew,1066
-likelihood-1.2.20.dist-info/METADATA,sha256=0Gw81rc3h8Ovr2ESuqRexKWQroFl9MUJMu3oi9wfAZw,2518
-likelihood-1.2.20.dist-info/WHEEL,sha256=P9jw-gEje8ByB7_hXoICnHtVCrEwMQh-630tKvQWehc,91
-likelihood-1.2.20.dist-info/top_level.txt,sha256=KDiBLr870YTxqLFqObTOSrTK10uw8dFsITSNLlte3PA,11
-likelihood-1.2.20.dist-info/RECORD,,
+likelihood-1.2.21.dist-info/LICENSE,sha256=XWHWt9egYEUHGPTnlcZfJKLPmysacOwdiLj_-J7Z9ew,1066
+likelihood-1.2.21.dist-info/METADATA,sha256=G3reOFrns927OS-FhndEABT5Xa2BJj8GTrNVOJ9F4fU,2518
+likelihood-1.2.21.dist-info/WHEEL,sha256=R06PA3UVYHThwHvxuRWMqaGcr-PuniXahwjmQRFMEkY,91
+likelihood-1.2.21.dist-info/top_level.txt,sha256=KDiBLr870YTxqLFqObTOSrTK10uw8dFsITSNLlte3PA,11
+likelihood-1.2.21.dist-info/RECORD,,

{likelihood-1.2.20.dist-info → likelihood-1.2.21.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (75.3.0)
+Generator: setuptools (75.5.0)
 Root-Is-Purelib: true
 Tag: py3-none-any