Update README.md

README.md

@@ -61,6 +61,39 @@ print(classification_report(y_test, y_pred))
 ```
 
 4. Training a new dataset with pre-trained model
-<br>To test a new dataset, combine the steps above：
-- 
+<br>To test a new dataset, follow the steps below：
+
+- Clean the Dataset
+```python
+from data_cleaning import clean
+
+# Load your dataset
+df = pd.read_csv('test_data_random_subset.csv')
+
+# Clean the data
+cleaned_df = clean(df)
+
+```
+
+- Extract TF-IDF Features
+```python
+from tfidf import tfidf
+
+# Transform the cleaned dataset
+X_new_tfidf = tfidf.transform(cleaned_df['title'])
+
+```
+
+- Make Predictions
+```python
+from svm import svm_model
+
+# Make predictions
+predictions = svm_model.predict(X_new_tfidf)
+
+# Add predictions to the dataset
+cleaned_df['predictions'] = predictions
+print(cleaned_df[['title', 'predictions']])
+
+```