Commit 057c6a72 authored by Ard Kastrati's avatar Ard Kastrati
Browse files

Merged the kerasTuner part

parent 8d1cc3e4
......@@ -82,11 +82,5 @@ def tune(trainX, trainY):
print(trainX.shape)
tuner.search_space_summary()
X_train, X_val, y_train, y_val = train_test_split(trainX, trainY, test_size=0.2, random_state=42)
<<<<<<< HEAD
tuner.search(X_train, y_train, epochs=1, validation_data=(X_val, y_val), verbose=1)
tuner.results_summary()
=======
tuner.search(X_train, y_train, epochs=15, validation_data=(X_val, y_val), verbose=2)
tuner.results_summary()
>>>>>>> 663f9249216598df283d8d9909058cf89bc05627
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