Skip to content

Commit

Permalink
Added attrs, although still needs work
Browse files Browse the repository at this point in the history
  • Loading branch information
DetectiveKiwi committed Apr 28, 2022
1 parent 190720b commit dd71484
Show file tree
Hide file tree
Showing 2 changed files with 28 additions and 1 deletion.
27 changes: 27 additions & 0 deletions nsdmd/nsdmd.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,33 @@
from scipy.stats import circmean

################## Classes
# from attrs import define
# @define
# class NSDMD:
# opt_win: int = 500
# opt_stride: int = 100
# opt_rank: int = 20
# bandpass: list[list[int]] = None
# bandpass_trim: int = 500
# sim_thresh_freq: float = 0.2
# sim_thresh_phi_amp: float = 0.95
# drift_N: int = 51
# exact_var_thresh: float = 0.01
# feature_init: float, int = None
# feature_N: int = 20
# feature_seq_method: str = 'SBS'
# feature_f_method: str = 'exact'
# feature_maxiter: int = 5
# feature_final_num: int = None
# feature_maxiter_float: int = 1
# grad_alpha: float = 0.1
# grad_beta: float = 0.1
# grad_N: int = 20
# grad_lr: float = 0.01
# grad_maxiter: int = 100
# grad_fit_coupling: bool = False
# verbose: bool = False


class NSDMD():
def __init__(self, opt_win=500, opt_stride=100, opt_rank=20, bandpass=None, bandpass_trim=500, \
Expand Down
2 changes: 1 addition & 1 deletion setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,5 +21,5 @@
"Topic :: Scientific/Engineering"
],
python_requires='>=3.8',
install_requires=["numpy","colorednoise","scipy"]
install_requires=["numpy","colorednoise","scipy","attrs"]
)

0 comments on commit dd71484

Please sign in to comment.