zhangyiss/mlx-examples
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx-examples.git synced 2025-06-24 01:17:28 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix import order of normalizing_flow (#326)

Browse Source
This commit is contained in:
AtomicVar 2024-01-17 00:45:55 +08:00 committed by GitHub
parent 2287294723
commit ce7b65e8c4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 6 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
normalizing_flow/distributions.py
View File

@ -1,7 +1,7 @@
# Copyright © 2023-2024 Apple Inc. # Copyright © 2023-2024 Apple Inc.
from typing import Tuple, Optional, Union
import math import math
from typing import Optional, Tuple, Union
import mlx.core as mx import mlx.core as mx

5
normalizing_flow/flows.py
View File

@ -1,11 +1,10 @@
# Copyright © 2023-2024 Apple Inc. # Copyright © 2023-2024 Apple Inc.
from typing import Tuple, Optional, Union from typing import Optional, Tuple, Union
import mlx.core as mx import mlx.core as mx
import mlx.nn as nn import mlx.nn as nn
from bijectors import AffineBijector, MaskedCoupling
from bijectors import MaskedCoupling, AffineBijector
from distributions import Normal from distributions import Normal

8
normalizing_flow/main.py
View File

@ -1,15 +1,13 @@
# Copyright © 2023-2024 Apple Inc. # Copyright © 2023-2024 Apple Inc.
from tqdm import trange
import numpy as np
from sklearn import datasets, preprocessing
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
import mlx.core as mx import mlx.core as mx
import mlx.nn as nn import mlx.nn as nn
import mlx.optimizers as optim import mlx.optimizers as optim
import numpy as np
from flows import RealNVP from flows import RealNVP
from sklearn import datasets, preprocessing
from tqdm import trange
def get_moons_dataset(n_samples=100_000, noise=0.06): def get_moons_dataset(n_samples=100_000, noise=0.06):