0
我試圖找到NN迴歸模型採用GridSearchCV與下面的代碼的最佳參數:如何使用GridSearchCV獲取每組參數的預測值?
param_grid = dict(optimizer=optimizer, epochs=epochs, batch_size=batches, init=init
grid = GridSearchCV(estimator=model, param_grid=param_grid, scoring='neg_mean_squared_error')
grid_result = grid.fit(input_train, target_train)
pred = grid.predict(input_test)
據我瞭解,grid.predict(input_test)
使用最佳參數來預測給定的輸入設置。有什麼方法可以使用測試集來評估每組參數的GridSearchCV嗎?
實際上,我的測試集包含一些特殊記錄,我想測試模型的一般性和準確性。謝謝。
謝謝@ eduard-ilyasov。它工作完美。 – saleh