Lab 4 Solution


Category: Tag:



  1. This note book contains instructions for COMP9318-Lab4.

  2. You are required to complete your implementation in a file provided along with this notebook.

  3. You are not allowed to print out unnecessary stuff. We will not consider any output printed out on the screen. All results should be returned in appropriate data structures via corresponding functions.

  4. You can submit your implementation for Lab4 via following link: .

  5. For each question, we have provided you with detailed instructions along with question headings. In case of any problem, you can post your query @ Piazza.

  6. You are allowed to add other functions and/or import modules (you may have to in this lab), but you are not allowed to define global variables. Only functions are allowed in

  7. You should not import unnecessary modules/libraries, failing to import such modules at test time will lead to errors.

  8. We will provide immediate feedback on your submission. You can access your scores using the online submission portal on the same day.

  9. For Final Evaluation we will be using a different dataset, so your final scores may vary.

  10. You are allowed to submit as many times as you want before the deadline, but ONLY the latest version will be kept and marked.

  11. Submission deadline for this assignment is 23:59:59 on 16th April, 2019. We will not accept any late submissions.

Question-1: Text Classification using Multinomial Naive Bayes

You are required to implement a multinomial naive bayes classifier to predict spam SMS.

The training data is a set of SMS categoried into spam and ham.

In [3]:

import pandas as pd

raw_data = pd.read_csv('./asset/data.txt', sep='\t')






Go until jurong point, crazy.. Available only …



Ok lar… Joking wif u oni…



Free entry in 2 a wkly comp to win FA Cup fina…



U dun say so early hor… U c already then say…



Nah I don’t think he goes to usf, he lives aro…

In order to implement a unigram model, first we tokenize the text. We use the count corresponding to each token (word) in the SMS as its feature (i.e., bag of words). We store the features and catrgorical information for each SMS in a dictionary.

In [4]:

def tokenize(sms):
    return sms.split(' ')

def get_freq_of_tokens(sms):
    tokens = {}
    for token in tokenize(sms):
        if token not in tokens:
            tokens[token] = 1
            tokens[token] += 1
    return tokens

training_data = []
for index in range(len(raw_data)):
    training_data.append((get_freq_of_tokens(raw_data.iloc[index].text), raw_data.iloc[index].category))

For this lab, you need to implement a multinomial naive bayes classifier (i.e., multinomial_nb() in the file: with add-1 smoothing. The input arguments of multinomial_nb() are:

  • training_data: pre-processed data stored as a dictionary

  • sms: test-sms (i.e., a list of tokens) that you need to categorize as spam and/or ham

The return value of multinomial_nb() should be the ratio of the probability of sms is spam and the probability of sms is ham. A return value larger than 1 implies the sms is spam and vice versa.

For example, a sample output is shown in the cell given below:

In [5]:

## How we test your implementation...
import submission

sms = 'I am not spam'
print(submission.multinomial_nb(training_data, tokenize(sms)))

Test Environment

For testing, we have pre-installed the requisite modules and/or libraries in the testing environment. You are only allowed to use following libraries:

  • python: 3.6.5

  • pandas: 0.19.2

NOTE: You are required to implement the classifier by yourself. You are not allowed to import sklearn and/or any other library in Lab4.

In [ ]:

error: Content is protected !!