Commit bfd156e3 authored by lucas_miranda's avatar lucas_miranda
Browse files

Fixed deepof.train_utils.tune_search

parent a0853105
Pipeline #103687 failed with stages
in 27 minutes and 41 seconds
......@@ -58,7 +58,7 @@ def plot_heatmap(
)
else:
sns.kdeplot(
data=heatmap.x, data2=heatmap.y, cmap=None, shade=True, alpha=1, ax=ax
x=heatmap.x, y=heatmap.y, cmap=None, shade=True, alpha=1, ax=ax
)
ax = np.array([ax])
......
......@@ -66,7 +66,6 @@ def test_get_callbacks(
rule_based_prediction=rule_based_prediction,
overlap_loss=overlap_loss,
loss=loss,
X_val=X_train,
input_type=False,
cp=True,
reg_cat_clusters=False,
......@@ -180,13 +179,11 @@ def test_tune_search(
next_sequence_prediction=np.round(next_sequence_prediction, 2),
rule_based_prediction=np.round(rule_based_prediction, 2),
loss=loss,
X_val=X_train,
input_type=False,
cp=False,
reg_cat_clusters=True,
reg_cluster_variance=True,
overlap_loss=overlap_loss,
entropy_samples=10,
entropy_knn=5,
logparam={"encoding": 2, "k": 15},
)
......
Markdown is supported
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