mimical 0.1.1__py3-none-any.whl → 0.1.2__py3-none-any.whl

mimical/fitting/fitter.py

@@ -14,11 +14,15 @@ from ..plotting import Plotter
 
 from ..utils import filter_set
 
+from tqdm import tqdm
+
+
 dir_path = os.getcwd()
 if not os.path.isdir(dir_path + "/mimical"):
     os.system('mkdir ' + dir_path + "/mimical")
     os.system('mkdir ' + dir_path + "/mimical/plots")
     os.system('mkdir ' + dir_path + "/mimical/posteriors")
+    os.system('mkdir ' + dir_path + "/mimical/cats")
 
 
 
@@ -162,6 +166,7 @@ class mimical(object):
             self.samples = pd.read_csv(dir_path+'/mimical/posteriors' + f'/{self.id}.txt', delimiter=' ').to_numpy()
             fit_dic = dict(zip((np.array((list(self.fitter_prior.keys)))+"_50").tolist(), np.median(self.samples, axis=0).tolist()))
             print(f"Loading existing posterior at " + dir_path + '/mimical/posteriors' + f'/{self.id}.txt')
+            self.save_cat()
             print(" ")
             return fit_dic
 
@@ -200,7 +205,35 @@ class mimical(object):
             print("Sampling finished successfully.")
             print(" ")
 
+            self.save_cat()
+
             return fit_dic
+        
+    
+    def save_cat(self):
+
+        user_samples = np.zeros((self.samples.shape[0], len(self.wavs), len(self.user_prior.keys())))
+        # Get median Nautilus parameters and transalte into median model parameters.
+        for j in tqdm(range(self.samples.shape[0])):
+            # Get median Nautilus parameters and transalte into median model parameters.
+            param_dict = dict(zip(list(self.fitter_prior.keys), self.samples[j]))
+            pars = self.prior_handler.revert(param_dict, self.wavs)
+            user_samples[j] = pars
+
+        quantiles = np.percentile(user_samples, q=(16, 50, 84), axis=0)
+
+        dic = {}
+        for j in range(len(self.filter_names)):
+            for i in range(len(self.user_prior.keys())):
+                key = list(self.user_prior.keys())[i]
+                dic[key + "_" + self.filter_names[j] + "_16"] = [quantiles[0, j, i]]
+                dic[key + "_" + self.filter_names[j] + "_50"] = [quantiles[1, j, i]]
+                dic[key + "_" + self.filter_names[j] + "_84"] = [quantiles[2, j, i]]
+
+        df = pd.DataFrame(dic)
+        df.to_csv(dir_path+'/mimical/cats' + f'/{self.id}.csv', index=False)
+
+
     
 
     def plot_model(self, type='median'):

{mimical-0.1.1.dist-info → mimical-0.1.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mimical
-Version: 0.1.1
+Version: 0.1.2
 Summary: Intesity modelling of multiply-imaged objects
 Author: Struan Stevenson
 Author-email: struan.stevenson@ed.ac.uk

{mimical-0.1.1.dist-info → mimical-0.1.2.dist-info}/RECORD

@@ -1,12 +1,12 @@
 mimical/__init__.py,sha256=cBuyDtPwn6KcIkjlT7oct36jigvYDJmbZNmnt1I1QJU,75
 mimical/fitting/__init__.py,sha256=xcnLoeBDtt4Le5NhD6GhJe2ZRPNSsEySRHFYNuJyNEQ,69
-mimical/fitting/fitter.py,sha256=fT9IdiiSIHQFQ_er7BWRT_YxQzxWboPvfFQtPKKIa6I,9186
+mimical/fitting/fitter.py,sha256=BjPT3wgGJ_KO4iyB9n0t4cqn0jfpU9d4IT0Gc1COoJc,10489
 mimical/fitting/prior_handler.py,sha256=yXvxbohS6f94CMRyeIViFc83hRYAj82IZNA5mzhMSRw,4193
 mimical/plotting/__init__.py,sha256=I1ZpQA48g-YPkqwrDGgTrqaTxDUgR8n0mbBX0MdbZtU,30
 mimical/plotting/plotting.py,sha256=y6uNeUh3Ifp8fapTPyN2eKT3OfnRhfNCUYq3ZN6lei4,6708
 mimical/utils/__init__.py,sha256=yolMgYFpvXoHuyV-ijGo26SQ0_niw0gz__iAFgdiQiY,35
 mimical/utils/filter_set.py,sha256=EITLa2c3FG3n1-v7KatKRwHCRyfZ4TTjsi_WM7Bz64k,6819
-mimical-0.1.1.dist-info/METADATA,sha256=C8vClVw6KON9OkjjOJuP5FW0jStMsQ0MRH6-u2duMBA,839
-mimical-0.1.1.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-mimical-0.1.1.dist-info/top_level.txt,sha256=z6HTYpsoNjLUFayXjn8WyjX8C1mIbZw8Arb334cUbwc,8
-mimical-0.1.1.dist-info/RECORD,,
+mimical-0.1.2.dist-info/METADATA,sha256=shzoz1K-wMSdIY0qzujhX7pxsoU6FBqcqy_3hRMgH1o,839
+mimical-0.1.2.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+mimical-0.1.2.dist-info/top_level.txt,sha256=z6HTYpsoNjLUFayXjn8WyjX8C1mIbZw8Arb334cUbwc,8
+mimical-0.1.2.dist-info/RECORD,,