mirror of
https://github.com/ArthurDanjou/handson-ml3.git
synced 2026-02-03 13:37:49 +01:00
In TF 2.2.0-rc1, validation_data expects tuples, not lists, fixes #131
This commit is contained in:
@@ -592,7 +592,7 @@
|
||||
"outputs": [],
|
||||
"source": [
|
||||
"model.compile(loss=\"sparse_categorical_crossentropy\", optimizer=\"nadam\", metrics=[\"accuracy\"])\n",
|
||||
"history = model.fit(X_train, y_train, epochs=10, validation_data=[X_valid, y_valid])\n",
|
||||
"history = model.fit(X_train, y_train, epochs=10, validation_data=(X_valid, y_valid))\n",
|
||||
"score = model.evaluate(X_test, y_test)\n",
|
||||
"X_new = X_test[:10] # pretend we have new images\n",
|
||||
"y_pred = model.predict(X_new)"
|
||||
@@ -1306,7 +1306,7 @@
|
||||
"model.compile(loss=\"sparse_categorical_crossentropy\", optimizer=\"nadam\",\n",
|
||||
" metrics=[\"accuracy\"])\n",
|
||||
"\n",
|
||||
"model.fit(X_train, y_train, epochs=10, validation_data=[X_valid, y_valid])\n",
|
||||
"model.fit(X_train, y_train, epochs=10, validation_data=(X_valid, y_valid))\n",
|
||||
"model.evaluate(X_test, y_test)"
|
||||
]
|
||||
},
|
||||
|
||||
Reference in New Issue
Block a user