Commit 305b1aba authored by lucas_miranda's avatar lucas_miranda
Browse files

Refactored a bit the old visualization pipeline - added umap-learn to requirements

parent e99ef7a8
Pipeline #87253 passed with stage
in 16 minutes and 18 seconds
......@@ -20,4 +20,4 @@ sklearn~=0.0
tensorflow-probability~=0.10.0
tensorflow~=2.2.0
tqdm~=4.50.0
umap~=0.1.1
umap-learn~=0.4.6
......@@ -265,6 +265,22 @@ if not val_num:
"Set a valid data path / validation number for the validation to run"
)
# Load checkpoints and build dataframe with predictions
path = checkpoints
checkpoints = sorted(
list(
set(
[
path + re.findall('(.*\.ckpt).data', i)[0]
for i in os.listdir(path)
if "ckpt.data" in i
]
)
)
)
assert len(checkpoints) >= 1, "Please provide a valid checkpoint path!"
assert input_type in [
"coords",
"dists",
......@@ -348,21 +364,6 @@ if pheno_class > 0:
print("Done!")
# Load checkpoints and build dataframe with predictions
path = checkpoints
checkpoints = sorted(
list(
set(
[
path + re.findall('(.*\.ckpt).data', i)[0]
for i in os.listdir(path)
if "ckpt.data" in i
]
)
)
)
pttest_idx = np.random.choice(list(range(X_train.shape[0])), samples)
pttest = X_train[pttest_idx]
......
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