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))
|
||||
|
||||
# Make predictions on new data
|
||||
new_writer = 'John Smith'
|
||||
new_writer = 'Jim Cash'
|
||||
new_writer_encoded = encoder.transform([new_writer])
|
||||
|
||||
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