Commit 73c2ca71 authored by mjennewine's avatar mjennewine
Browse files

Slight performance increase using one-hot

parent 8c431c32

X0.joblib

(76.3 MiB)

File changed.

No diff preview for this file type.

View original file
View changed file

X1.joblib

(81.5 MiB)

File changed.

No diff preview for this file type.

View original file
View changed file

nn.joblib

−155 B (3.07 MiB)

File changed.

No diff preview for this file type.

View original file
View changed file

pla.joblib

(8.91 KiB)

File changed.

No diff preview for this file type.

View original file
View changed file

preprocess.py

+1 −2
Original line number Diff line number Diff line
@@ -102,7 +102,6 @@ def extract(message): 
    #create list of words from text

    words = clean.split()

    

    """

    #one-hot

    #set feature value to 1 if word is found in text

    for i in range(len(features)):
@@ -112,7 +111,7 @@ def extract(message): 
    #count occurrences of feature word in text

    for i in range(len(features)):

        vector[i] = words.count(features[i])

    

    """

    return vector