upd52
This commit is contained in:
parent
26a2ed79b8
commit
df3eba461b
@ -34,8 +34,8 @@ model.compile(loss='mean_squared_error', optimizer='adam')
|
|||||||
model.fit(X_train, y_train, batch_size=64, epochs=10, validation_data=(X_test, y_test))
|
model.fit(X_train, y_train, batch_size=64, epochs=10, validation_data=(X_test, y_test))
|
||||||
|
|
||||||
# Make predictions on new data
|
# Make predictions on new data
|
||||||
new_writer = 'John Smith'
|
new_writer = 'Jim Cash'
|
||||||
new_writer_encoded = encoder.transform([new_writer])
|
new_writer_encoded = encoder.transform([new_writer])
|
||||||
|
|
||||||
rating_prediction = model.predict(new_writer_encoded)
|
rating_prediction = model.predict(new_writer_encoded)
|
||||||
print("Predicted rating for the writer 'John Smith':", rating_prediction)
|
print("Predicted rating for the writer 'Jim Cash':", rating_prediction)
|
Loading…
Reference in New Issue
Block a user