Commit 30215fba authored by syz's avatar syz
Browse files

First tested version of the Bayesian Inference class!

parent 27523a83
......@@ -38,6 +38,7 @@ from .feature_extraction import FeatureExtractorParallel, FeatureExtractorSerial
from .image_transformation import geoTransformerParallel, geoTransformerSerial
from . import process
from .process import Process
from .giv_bayesian import GIVBayesian
def no_impl(*args, **kwargs):
......@@ -57,4 +58,4 @@ else:
geoTransformer = geoTransformerParallel
__all__ = ['Cluster', 'Decomposition', 'ImageWindow', 'doSVD', 'fft', 'gmode_utils', 'proc_utils', 'svd_utils',
'giv_utils', 'rebuild_svd', 'Process', 'parallel_compute']
'giv_utils', 'rebuild_svd', 'Process', 'parallel_compute', 'Process', 'GIVBayesian']
......@@ -22,7 +22,7 @@ cap_dtype = np.dtype({'names': ['Forward', 'Reverse'],
'formats': [np.float32, np.float32]})
class GIVbayesian(Process):
class GIVBayesian(Process):
def __init__(self, h5_main, ex_freq, gain, num_x_steps=250, r_extra=220, verbose=False,
cores=1, max_mem_mb=1024, **kwargs):
"""
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment