Update comment_analyzer.py

comment_analyzer.py

@@ -12,27 +12,22 @@ data = pd.read_csv("modeled_data.csv")
 analyzer = SentimentIntensityAnalyzer()
 
 
-def sample_model(df, regressor, scale=None):
+def sample_model(df, regressor):
     X = df.drop("rate",axis=1)
     y = df["rate"]
     
     X_train, X_test, y_train, y_test = train_test_split(X, y, test_size = 0.2, random_state=1)
-    scaled_X_train, scaled_X_test = X_train, X_test
-    if scale != None:
-        scaler = scale
-        scaled_X_train = pd.DataFrame(scaler.fit_transform(X_train), columns = X_train.columns)
-        scaled_X_test = pd.DataFrame(scaler.transform(X_test),columns = X_test.columns)
     
     model = regressor
-    model.fit(scaled_X_train, y_train)
-    y_pred = model.predict(scaled_X_test)
+    model.fit(X_train, y_train)
+    y_pred = model.predict(X_test)
     
     rmse = np.sqrt(mean_squared_error(y_test, y_pred))
     
-    return model, scaled_X_train, scaled_X_test, y_train, y_test 
+    return model
 
 
-def user_interaction(comment, model):
+def calculate_sentiments(comment, model):
     
     negative_score = analyzer.polarity_scores(comment)["neg"]
     neutral_score = analyzer.polarity_scores(comment)["neu"]
@@ -45,8 +40,8 @@ def user_interaction(comment, model):
     
 def take_input(comment):
     
-    cons_tuned_svr, _, _, _, _ = sample_model(data, SVR(C=3, kernel="rbf", tol=0.001))
-    return user_interaction(comment, cons_tuned_svr)
+    cons_tuned_svr = sample_model(data, SVR(C=3, kernel="rbf", tol=0.001))
+    return calculate_sentiments(comment, cons_tuned_svr)
 
 
 with gr.Blocks() as demo:
@@ -57,8 +52,8 @@ with gr.Blocks() as demo:
                     ##### Thanks for your interest and taking your time.
                     ##### Tell us about your personal experience enrolling in this course. Was it the right match for you?
                     """)
-        input_comment = gr.Textbox(placeholder="Write your comment here...", show_label = False)
-        button = gr.Button("What is the Rating I Have Given? Click me to Learn")
+        input_comment = gr.Textbox(placeholder="Write your comment here...", show_label = False, lines=2)
+        button = gr.Button("What is the Rating I Have Given? Click me to Learn", variant="primary").style(full_width=True)
         with gr.Row():
             with gr.Column():
                 gr.Markdown("#### Generated Rating from Your Comment")