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( ...@@ -58,7 +58,7 @@ def plot_heatmap(
) )
else: else:
sns.kdeplot( 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]) ax = np.array([ax])
......
...@@ -66,7 +66,6 @@ def test_get_callbacks( ...@@ -66,7 +66,6 @@ def test_get_callbacks(
rule_based_prediction=rule_based_prediction, rule_based_prediction=rule_based_prediction,
overlap_loss=overlap_loss, overlap_loss=overlap_loss,
loss=loss, loss=loss,
X_val=X_train,
input_type=False, input_type=False,
cp=True, cp=True,
reg_cat_clusters=False, reg_cat_clusters=False,
...@@ -180,13 +179,11 @@ def test_tune_search( ...@@ -180,13 +179,11 @@ def test_tune_search(
next_sequence_prediction=np.round(next_sequence_prediction, 2), next_sequence_prediction=np.round(next_sequence_prediction, 2),
rule_based_prediction=np.round(rule_based_prediction, 2), rule_based_prediction=np.round(rule_based_prediction, 2),
loss=loss, loss=loss,
X_val=X_train,
input_type=False, input_type=False,
cp=False, cp=False,
reg_cat_clusters=True, reg_cat_clusters=True,
reg_cluster_variance=True, reg_cluster_variance=True,
overlap_loss=overlap_loss, overlap_loss=overlap_loss,
entropy_samples=10,
entropy_knn=5, entropy_knn=5,
logparam={"encoding": 2, "k": 15}, 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