diff --git a/script5_2.py b/script5_2.py index f006cd5..96f340f 100644 --- a/script5_2.py +++ b/script5_2.py @@ -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) \ No newline at end of file +print("Predicted rating for the writer 'Jim Cash':", rating_prediction) \ No newline at end of file