I have a neural network model (model.predict in code) which fitted our data, and now we want optimized some variables that exist in the model. my code are attached below,
def objective_function(X):
wob = X[0]
torque= X[1]
RPM = X[2]
pump = X[3]
input = pd.DataFrame(wob,torque,RPM, 0.00017,0.027,pump,0,0.5,0.386,0.026,0.0119,0.33,0.83,0.48)
obj= model.predict(input)
return obj
I want to know my function code is correct or not?
What I have tried:
you can see the code, I attached the code. I'm not expert in python, unfortunately.