-
-
Notifications
You must be signed in to change notification settings - Fork 8.3k
Expand file tree
/
Copy pathmlab.pyi
More file actions
100 lines (95 loc) · 3.5 KB
/
mlab.pyi
File metadata and controls
100 lines (95 loc) · 3.5 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
from collections.abc import Callable
import functools
from typing import Literal
import numpy as np
from numpy.typing import ArrayLike
def window_hanning(x: ArrayLike) -> ArrayLike: ...
def window_none(x: ArrayLike) -> ArrayLike: ...
def detrend(
x: ArrayLike,
key: Literal["default", "constant", "mean", "linear", "none"]
| Callable[[ArrayLike, int | None], ArrayLike]
| None = ...,
axis: int | None = ...,
) -> ArrayLike: ...
def detrend_mean(x: ArrayLike, axis: int | None = ...) -> ArrayLike: ...
def detrend_none(x: ArrayLike, axis: int | None = ...) -> ArrayLike: ...
def detrend_linear(y: ArrayLike) -> ArrayLike: ...
def psd(
x: ArrayLike,
NFFT: int | None = ...,
Fs: float | None = ...,
detrend: Literal["none", "mean", "linear"]
| Callable[[ArrayLike, int | None], ArrayLike]
| None = ...,
window: Callable[[ArrayLike], ArrayLike] | ArrayLike | None = ...,
noverlap: int | None = ...,
pad_to: int | None = ...,
sides: Literal["default", "onesided", "twosided"] | None = ...,
scale_by_freq: bool | None = ...,
) -> tuple[ArrayLike, ArrayLike]: ...
def csd(
x: ArrayLike,
y: ArrayLike | None,
NFFT: int | None = ...,
Fs: float | None = ...,
detrend: Literal["none", "mean", "linear"]
| Callable[[ArrayLike, int | None], ArrayLike]
| None = ...,
window: Callable[[ArrayLike], ArrayLike] | ArrayLike | None = ...,
noverlap: int | None = ...,
pad_to: int | None = ...,
sides: Literal["default", "onesided", "twosided"] | None = ...,
scale_by_freq: bool | None = ...,
) -> tuple[ArrayLike, ArrayLike]: ...
complex_spectrum = functools.partial(tuple[ArrayLike, ArrayLike])
magnitude_spectrum = functools.partial(tuple[ArrayLike, ArrayLike])
angle_spectrum = functools.partial(tuple[ArrayLike, ArrayLike])
phase_spectrum = functools.partial(tuple[ArrayLike, ArrayLike])
def specgram(
x: ArrayLike,
NFFT: int | None = ...,
Fs: float | None = ...,
detrend: Literal["none", "mean", "linear"] | Callable[[ArrayLike, int | None], ArrayLike] | None = ...,
window: Callable[[ArrayLike], ArrayLike] | ArrayLike | None = ...,
noverlap: int | None = ...,
pad_to: int | None = ...,
sides: Literal["default", "onesided", "twosided"] | None = ...,
scale_by_freq: bool | None = ...,
mode: Literal["psd", "complex", "magnitude", "angle", "phase"] | None = ...,
) -> tuple[ArrayLike, ArrayLike, ArrayLike]: ...
def cohere(
x: ArrayLike,
y: ArrayLike,
NFFT: int = ...,
Fs: float = ...,
detrend: Literal["none", "mean", "linear"] | Callable[[ArrayLike, int | None], ArrayLike] = ...,
window: Callable[[ArrayLike], ArrayLike] | ArrayLike = ...,
noverlap: int = ...,
pad_to: int | None = ...,
sides: Literal["default", "onesided", "twosided"] = ...,
scale_by_freq: bool | None = ...,
) -> tuple[ArrayLike, ArrayLike]: ...
class GaussianKDE:
dataset: ArrayLike
dim: int
num_dp: int
factor: float
data_covariance: ArrayLike
data_inv_cov: ArrayLike
covariance: ArrayLike
inv_cov: ArrayLike
norm_factor: float
def __init__(
self,
dataset: ArrayLike,
bw_method: Literal["scott", "silverman"]
| float
| Callable[[GaussianKDE], float]
| None = ...,
) -> None: ...
def scotts_factor(self) -> float: ...
def silverman_factor(self) -> float: ...
def covariance_factor(self) -> float: ...
def evaluate(self, points: ArrayLike) -> np.ndarray: ...
def __call__(self, points: ArrayLike) -> np.ndarray: ...