fork download
  1. # Importing necessary libraries
  2. import pandas as pd
  3. from sklearn.feature_extraction.text import TfidfVectorizer
  4. from sklearn.model_selection import train_test_split
  5. from sklearn.linear_model import LogisticRegression
  6. from sklearn.metrics import accuracy_score
  7.  
  8. # Sample dataset (replace with your dataset)
  9. data = pd.DataFrame({
  10. 'text': ["I love this movie!", "This movie is terrible.", "Neutral tweet about something."],
  11. 'sentiment': ['positive', 'negative', 'neutral']
  12. })
  13.  
  14. # Split data into features (X) and target (y)
  15. X = data['text']
  16. y = data['sentiment']
  17.  
  18. # Vectorize text data using TF-IDF representation
  19. vectorizer = TfidfVectorizer()
  20. X_vect = vectorizer.fit_transform(X)
  21.  
  22. # Split data into training and testing sets
  23. X_train, X_test, y_train, y_test = train_test_split(X_vect, y, test_size=0.2, random_state=42)
  24.  
  25. # Build and train Logistic Regression model
  26. model = LogisticRegression()
  27. model.fit(X_train, y_train)
  28.  
  29. # Evaluate model
  30. y_pred = model.predict(X_test)
  31. accuracy = accuracy_score(y_test, y_pred)
  32.  
  33. print("Accuracy:", accuracy)
  34.  
Success #stdin #stdout 0.79s 105488KB
stdin


stdout
Accuracy: 0.0