yitingliii
commited on
Commit
•
b544376
1
Parent(s):
6b2dc18
Update README.md
Browse files
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 |
-
|
58 |
df = pd.read_csv("hf://datasets/CIS5190abcd/headlines_test/test_cleaned_headlines.csv")
|
59 |
|
60 |
-
|
61 |
cleaned_df = clean(df)
|
62 |
|
63 |
```
|
@@ -66,14 +66,14 @@ cleaned_df = clean(df)
|
|
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 |
```
|
|
|
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 |
```
|