Skip to content

Commit 8d3c63e

Browse files
committed
mtd_kde default tll; fastkde==2.1.3
1 parent a671595 commit 8d3c63e

File tree

3 files changed

+8
-10
lines changed

3 files changed

+8
-10
lines changed

.gitignore

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,8 @@ MNIST/
1111
logs/
1212
examples/MNIST
1313
examples/logs
14+
examples/**/*.gz
15+
examples/**/*.ipynb
1416

1517
# Python-generated files
1618
__pycache__/

pyproject.toml

Lines changed: 4 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -25,13 +25,7 @@ maintainers = [
2525
]
2626
description = "yet another vine copula library, using PyTorch."
2727
readme = "README.md"
28-
keywords = [
29-
"vine copula",
30-
"copula",
31-
"torch",
32-
"conditional simulation",
33-
"fastkde",
34-
]
28+
keywords = ["vine copula", "copula", "torch", "conditional simulation"]
3529
classifiers = [
3630
# "Private :: Do Not Upload",
3731
"License :: OSI Approved :: MIT License",
@@ -44,10 +38,12 @@ classifiers = [
4438
"Programming Language :: Python :: 3.13",
4539
"Topic :: Scientific/Engineering",
4640
]
47-
dependencies = ["numpy>=2", "scipy", "fastkde", "pyvinecopulib"]
41+
dependencies = ["numpy>=2", "scipy", "fastkde==2.1.3", "pyvinecopulib"]
4842

4943
[dependency-groups]
5044
dev = [
45+
"ccxt",
46+
"missingno",
5147
"blitz-bayesian-pytorch",
5248
"coverage",
5349
"datasets",

torchvinecopulib/vinecop/__init__.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -417,7 +417,7 @@ def fit(
417417
mtd_vine: str = "rvine",
418418
mtd_bidep: str = "chatterjee_xi",
419419
thresh_trunc: None | float = 0.01,
420-
mtd_kde: str = "fastKDE",
420+
mtd_kde: str = "tll",
421421
mtd_tll: str = "constant",
422422
num_iter_max: int = 17,
423423
is_tau_est: bool = False,
@@ -434,7 +434,7 @@ def fit(
434434
matrix (torch.Tensor, optional): matrix representation of the vine structure. Defaults to None.
435435
first_tree_vertex (tuple, optional): vertices of the first tree (set of conditioning variables). Defaults to ().
436436
mtd_vine (str, optional): method for vine structure. One of "cvine", "dvine", "rvine". Defaults to "rvine".
437-
mtd_bidep (str, optional): method for bivariate dependence. One of "chatterjee_xi", "ferreira_tail_dep_coeff", "kendall_tau", "mutual_info", "spearman_rho". Defaults to "chatterjee_xi".
437+
mtd_bidep (str, optional): method for bivariate dependence. One of "chatterjee_xi", "ferreira_tail_dep_coeff", "kendall_tau", "mutual_info". Defaults to "chatterjee_xi".
438438
thresh_trunc (None | float, optional): threshold for truncating bivariate copulas using p-val from Kendall's tau stats test. Defaults to 0.01.
439439
mtd_kde (str, optional): method for bicop KDE. One of "fastKDE" or "tll". Defaults to "fastKDE".
440440
mtd_tll (str, optional): fit method for the transformation local-likelihood (TLL) nonparametric family, one of ("constant", "linear", or "quadratic"). Defaults to "constant".

0 commit comments

Comments
 (0)