Skip to content

Commit

Permalink
Removing _make_predict_function since it is not in tensorflow 2.4
Browse files Browse the repository at this point in the history
  • Loading branch information
joshua-cogliati-inl committed Jun 29, 2021
1 parent 90ebf62 commit d5c91f2
Showing 1 changed file with 0 additions and 1 deletion.
1 change: 0 additions & 1 deletion framework/SupervisedLearning/KerasBase.py
Original file line number Diff line number Diff line change
Expand Up @@ -449,7 +449,6 @@ def __trainLocal__(self,featureVals,targetVals):
#output layer
self._addOutputLayers()
self._ROM.compile(loss=self.lossFunction, optimizer=self.optimizer, metrics=self.metrics)
self._ROM._make_predict_function() # have to initialize before threading
self._romHistory = self._ROM.fit(featureVals, targetVals, epochs=self.epochs, batch_size=self.batchSize, validation_split=self.validationSplit)
# The following requires pydot-ng and graphviz to be installed (See the manual)
# https://github.com/keras-team/keras/issues/3210
Expand Down

0 comments on commit d5c91f2

Please sign in to comment.