A method of creating and using a neural network model for wellbore operations is disclosed. The method, in one aspect, may include defining a plurality of a wellbore parameter; calculating a plurality of output values of a tool operating parameter using the plurality of values of the wellbore parameter as input to a preexisting model; and obtaining a neural network model by using the plurality of values of the wellbore parameter and the calculated plurality of output values of the tool operating parameter. The neural network may be utilized for any suitable wellbore operation, including in conjunction with a drilling assembly for drilling a wellbore.