Databend 开源周报第 147 期

-- Create the sentiment analysis function
CREATE OR REPLACE FUNCTION sentiment_analysis(STRING) RETURNS STRING
LANGUAGE python HANDLER = 'sentiment_analysis'
AS $$
def sentiment_analysis(text):
    """
    Analyzes the sentiment of the input text by removing stop words and calculating the sentiment score.

    Args:
    text (str): The input text.

    Returns:
    str: Sentiment analysis result including the score and label.
    """
    stop_words = {"a", "an", "the", "and", "or", "but", "if", "then", "so"}
    positive_words = {"good", "happy", "joy", "excellent", "positive", "love"}
    negative_words = {"bad", "sad", "pain", "terrible", "negative", "hate"}

    # Remove stop words and calculate sentiment score
    words = [word for word in text.split() if word.lower() not in stop_words]
    score = sum((1 if word.lower() in positive_words else -1 if word.lower() in negative_words else 0) for word in words)

    # Determine sentiment label
    if score > 0:
        label = 'Positive'
    elif score < 0:
        label = 'Negative'
    else:
        label = 'Neutral'

    return f'Sentiment Score: {score}; Sentiment Label: {label}'
$$;

CREATE OR REPLACE TABLE texts (
    original_text STRING
);

-- Insert sample data
INSERT INTO texts (original_text)
VALUES
('The quick brown fox feels happy and joyful'),
('A hard journey, but it was painful and sad'),
('Uncertain outcomes leave everyone unsure and hesitant'),
('The movie was excellent and everyone loved it'),
('A terrible experience that made me feel bad');

-- Query to process the table data
SELECT
    original_text,
    sentiment_analysis(original_text) AS processed_text
FROM
    texts;