From 3949f9f433823cbe698b7ec29edc691ac367aaed Mon Sep 17 00:00:00 2001
From: Kevin Kadak <kada4470@mylaurier.ca>
Date: Wed, 13 Mar 2024 16:46:41 -0400
Subject: [PATCH] Docstrings added.

---
 whobpyt/datatypes/AbstractLoss.py | 91 ++++++++++++++++++-------------
 1 file changed, 52 insertions(+), 39 deletions(-)

diff --git a/whobpyt/datatypes/AbstractLoss.py b/whobpyt/datatypes/AbstractLoss.py
index 8fc60ee2..2bcf982d 100644
--- a/whobpyt/datatypes/AbstractLoss.py
+++ b/whobpyt/datatypes/AbstractLoss.py
@@ -1,39 +1,52 @@
-"""
-Authors: Andrew Clappison, John Griffiths, Zheng Wang, Davide Momi, Sorenza Bastiaens, Parsa Oveisi, Kevin Kadak, Taha Morshedzadeh, Shreyas Harita
-"""
-
-import torch
-from whobpyt.datatypes.parameter import par
-
-class AbstractLoss:
-    # This is the abstract class for objective function components, or for a custom objective function with multiple components. 
-
-    def __init__(self, simKey = None, model = None, device = torch.device('cpu')):
-    
-        self.simKey = simKey #This is a string key to extract from the dictionary of simulation outputs the time series used by the objective function
-        self.device =  device
-        self.model = model
-    
-    def main_loss(self, simData, empData):
-        # Calculates a loss to be backpropagated through
-        # If the objective function needs additional info, it should be defined at initialization so that the parameter fitting paradigms don't need to change
-        
-        # simData: is a dictionary of simulated state variable/neuroimaging modality time series. Typically accessed as simData[self.simKey].
-        # empData: is the target either as a time series or a calculated phenomena metric
-        
-        pass
-    
-    def prior_loss(self):
-        loss_prior = []
-        lb =0.001
-        m = torch.nn.ReLU()
-        variables_p = [a for a in dir(self.model.params) if type(getattr(self.model.params, a)) == par]
-        # get penalty on each model parameters due to prior distribution
-        for var_name in variables_p:
-            # print(var)
-            var = getattr(self.model.params, var_name)
-            if var.fit_hyper:
-                loss_prior.append(torch.sum((lb + m(var.prior_precision)) * \
-                                                (m(var.val) - m(var.prior_mean)) ** 2) \
-                                      + torch.sum(-torch.log(lb + m(var.prior_precision)))) #TODO: Double check about converting _v_inv 
-        return loss_prior
+"""
+Authors: Andrew Clappison, John Griffiths, Zheng Wang, Davide Momi, Sorenza Bastiaens, Parsa Oveisi, Kevin Kadak, Taha Morshedzadeh, Shreyas Harita
+"""
+
+import torch
+from whobpyt.datatypes.parameter import par
+
+class AbstractLoss:
+    """
+    This is the abstract class for objective function components, or for a custom objective function with multiple components.
+    
+    Attributes
+    ------------
+    simKey : string
+        String key to extract from the dictionary of simulation outputs the time series used by the objective function 
+    model : class
+        Class object for selected model type to be used
+    device : string ['cpu', 'gpu']
+        Defines which hardware component upon which to run the process
+
+    simData : dict of torch.Tensor with node_size X datapoint
+        Dictionary of simulated state variable/neuroimaging modality time series. Typically accessed as simData[self.simKey]
+    empData : torch.Tensor with node_size X datapoint
+        Target either as a time series or a calculated phenomena metric
+    """
+
+    def __init__(self, simKey = None, model = None, device = torch.device('cpu')):
+    
+        self.simKey = simKey
+        self.device =  device
+        self.model = model
+    
+    def main_loss(self, simData, empData):
+        # Calculates a loss to be backpropagated through
+        # If the objective function needs additional info, it should be defined at initialization so that the parameter fitting paradigms don't need to change
+        
+        pass
+    
+    def prior_loss(self):
+        loss_prior = []
+        lb = 0.001
+        m = torch.nn.ReLU()
+        variables_p = [a for a in dir(self.model.params) if type(getattr(self.model.params, a)) == par]
+        # get penalty on each model parameters due to prior distribution
+        for var_name in variables_p:
+            # print(var)
+            var = getattr(self.model.params, var_name)
+            if var.fit_hyper:
+                loss_prior.append(torch.sum((lb + m(var.prior_precision)) * \
+                                                (m(var.val) - m(var.prior_mean)) ** 2) \
+                                      + torch.sum(-torch.log(lb + m(var.prior_precision)))) #TODO: Double check about converting _v_inv 
+        return loss_prior