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 |
```
|