1

I'm using Autokeras to find the best regression model and I need to plot the learning curves of the best model. However, I cannot find the trial ID of the best model. Here is a part of my code:

#path for saving the logs

import datetime
%load_ext tensorboard
path = "logs/fit/" + datetime.datetime.now().strftime("%Y%m%d-%H%M%S")
tensorboard_callback= tf.keras.callbacks.TensorBoard(log_dir=path)

# Creating Keras Search Type
regressor = ak.StructuredDataRegressor(
                                        output_dim=2,
                                        loss="mean_squared_error",
                                        metrics=["mae","mean_squared_error"],
                                        project_name="structured_data_regressor",
                                        max_trials=3,
                                        directory=None,
                                        objective="val_mean_squared_error",
                                        tuner=None,
                                        overwrite=False,
                                        seed=100,
                                       )
#Fitting ANN to Training Set - Validation Data Provide --> Validation Split = 0.1111

history = regressor.fit(x = X_train, y = y_train, epochs= 5, callbacks=[tensorboard_callback], validation_split=0.2) 

#Learning Curves
%tensorboard --logdir logs/fit
Rita
  • 11
  • 1

0 Answers0