yitingliii commited on
Commit
b544376
1 Parent(s): 6b2dc18

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -4
README.md CHANGED
@@ -54,10 +54,10 @@ nltk.download('stopwords')
54
  ```
55
  <br> You can replace with any datasets you want by changing the file name inside ```pd.read_csv()```.
56
  ```python
57
- # Load your data
58
  df = pd.read_csv("hf://datasets/CIS5190abcd/headlines_test/test_cleaned_headlines.csv")
59
 
60
- # Clean the data
61
  cleaned_df = clean(df)
62
 
63
  ```
@@ -66,14 +66,14 @@ cleaned_df = clean(df)
66
  ```python
67
  from tfidf import tfidf
68
 
69
- # Transform the cleaned dataset
70
  X_new_tfidf = tfidf.transform(cleaned_df['title'])
71
 
72
  ```
73
 
74
  - Make Predictions
75
  ```python
76
- # Make predictions.
77
  from svm import svm_model
78
 
79
  ```
 
54
  ```
55
  <br> You can replace with any datasets you want by changing the file name inside ```pd.read_csv()```.
56
  ```python
57
+
58
  df = pd.read_csv("hf://datasets/CIS5190abcd/headlines_test/test_cleaned_headlines.csv")
59
 
60
+
61
  cleaned_df = clean(df)
62
 
63
  ```
 
66
  ```python
67
  from tfidf import tfidf
68
 
69
+
70
  X_new_tfidf = tfidf.transform(cleaned_df['title'])
71
 
72
  ```
73
 
74
  - Make Predictions
75
  ```python
76
+
77
  from svm import svm_model
78
 
79
  ```