Analysis

Suppose our text data is currently arranged into a single file, where each line of that file contains all of the text in a single document. Here we can use SFrame.read_csv to parse the text data into a one-column SFrame.

import os
if os.path.exists('wikipedia_w16'):
    sf = graphlab.SFrame('wikipedia_w16')
else:
    sf = graphlab.SFrame.read_csv('https://static.turi.com/datasets/wikipedia/raw/w16.csv', header=False)
    sf.save('wikipedia_w16')
sf
Columns:
X1      str

Rows: 72269

Data:
+--------------------------------+
|               X1               |
+--------------------------------+
| alainconnes alain connes i ... |
| americannationalstandardsi ... |
| alberteinstein near the be ... |
| austriangerman as german i ... |
| arsenic arsenic is a metal ... |
| alps the alps alpen alpi a ... |
| alexiscarrel born in saint ... |
| adelaide adelaide is a coa ... |
| artist an artist is a pers ... |
| abdominalsurgery the three ... |
|              ...               |
+--------------------------------+
[72269 rows x 1 columns]
Note: Only the head of the SFrame is printed.
You can use print_rows(num_rows=m, num_columns=n) to print more rows and columns.
Bag-of-words

Both SFrames and SArrays expose functionality that can be very useful for manipulating text data. For example, one common preprocessing task for text data is to transform it into "bag-of-words" format: each document is represented by a map where the words are keys and the values are the number of occurrences. So a document containing the text "hello goodbye hello" would be represented by a dict type element containing the value {"hello": 2, "goodbye":1}. This transformation can be accomplished with the following code.

bow = graphlab.text_analytics.count_words(sf['X1'])

We can print five of the words in the first document

bow[0].keys()[:5]
['and', 'work', 'baumconnes', 'gold', 'almost']

and find the documents that contain the word "gold":

bow.dict_has_any_keys(['gold'])

We can save this representation of the documents as another column of the original SFrame.

sf['bow'] = bow
TF-IDF

Another useful representation for text data is called TF-IDF (term frequency - inverse document frequency). This is a modification of the bag-of-words format where the counts are transformed into scores: words that are common across the document corpus are given low scores, and rare words occurring often in a document are given high scores.

where N(w, d) is the number of times word w occurs in document d. This transformation can be done to an SArray of dict type containing documents in bow-of-words format using tf_idf.

sf['tfidf'] = graphlab.text_analytics.tf_idf(sf['bow'])
BM25

The BM25 score is yet another useful representation for text data. It scores each document in a corpus according to the document's relevance to a particular query. For a query with terms , the BM25 score for document is:

where:

  • is the number of times term occurs in document ,
  • is the number of words in document ,
  • is the average number of words per document,
  • and are free parameters for Okapi BM25,

The first quantity in the sum is the inverse document frequency. For a corpus with documents, inverse document frequency for term is:

where is the number of documents in the corpus that contain term .

The transformed output is a column of type float with the BM25 score for each document. For more details on the BM25 score see http://en.wikipedia.org/wiki/Okapi_BM25.

query = ['beatles', 'john', 'paul']
bm25_scores = graphlab.text_analytics.bm25(dataset, query)
Text cleaning

We can easily remove all words do not occur at least twice in each document using SArray.dict_trim_by_values.

docs = sf['bow'].dict_trim_by_values(2)

Alternatively, we can remove all words which do not occur at least threshold number of times using the RareWordTrimmer.

GraphLab Create also contains a helper function called stopwords that returns a list of common words. We can use SArray.docs.dict_trim_by_keys to remove these words from the documents as a preprocessing step. NB: Currently only English words are available.

docs = docs.dict_trim_by_keys(graphlab.text_analytics.stopwords(), exclude=True)

To confirm that we have indeed removed common words, e.g. "and", "the", etc, we can examine the first document.

docs[0]
{'academy': 5,
 'algebras': 2,
 'connes': 3,
 'differential': 2,
 'early': 2,
 'geometry': 2,
 'including': 2,
 'medal': 2,
 'operator': 2,
 'physics': 2,
 'sciences': 5,
 'theory': 2,
 'work': 2}
Tokenization

For an SArray of strings, where each row is assumed to be a natural English language document, the tokenizer transforms each row into an ordered list of strings that represents the a simpler version of the Penn-Tree-Bank-style (PTB-style) tokenization of that row's document. For many text analytics tasks that require word-level granularity, simple space delimitation does not address some of the subtleties of natural language text, especially with respect to contractions, sentence-final punctuation, URL's, email addresses, phone numbers, and other quirks. The representation of a document provided by PTB-style of tokenization is essential for sequence-tagging, parsing, bag-of-words treatment, and any text analytics task that requires word-level granularity. For a description of this style of tokenization, see https://www.cis.upenn.edu/~treebank/tokenization.html.

tokenized_docs = graphlab.SFrame()
tokenized_docs['tokens'] = graphlab.text_analytics.tokenize(sf['X1'])
tokenized_docs
Columns:
  tokens  list

Rows: 72269

Data:
+-------------------------------+
|             tokens            |
+-------------------------------+
| [alainconnes, alain, conne... |
| [americannationalstandards... |
| [alberteinstein, near, the... |
| [austriangerman, as, germa... |
| [arsenic, arsenic, is, a, ... |
| [alps, the, alps, alpen, a... |
| [alexiscarrel, born, in, s... |
| [adelaide, adelaide, is, a... |
| [artist, an, artist, is, a... |
| [abdominalsurgery, the, th... |
+-------------------------------+
[72269 rows x 1 columns]
Note: Only the head of the SFrame is printed.
You can use print_rows(num_rows=m, num_columns=n) to print more rows and columns.

Note that our tokenizer does not normalize quote and bracket-like characters as described by the linked document.

Part of Speech Extraction

It can be useful to extract particular parts of speech. Specifically, you may want to highlight unique nouns in your text, identify adjectives with the high sentiment scores, or pull out nouns to generate candidate entities. The extract_parts_of_speech method parses the text in each element and extracts the words that are a given part of speech. For instance, to find all instances of adjectives:

parts_of_speech = graphlab.SFrame()
parts_of_speech['adjectives'] = graphlab.text_analytics.extract_parts_of_speech(sf['X1'],chosen_pos=[graphlab.text_analytics.PartOfSpeech.ADJ])
parts_of_speech
Columns:
  adjectives  dict

Rows: 10

Data:
+-------------------------------+
|           adjectives          |
+-------------------------------+
| {'ADJ': {'first': 1, 'nati... |
| {'ADJ': {'first': 2, 'tech... |
| {'ADJ': {'standard': 2, 'm... |
| {'ADJ': {'standard': 8, 'p... |
| {'ADJ': {'arsenopyrite': 2... |
| {'ADJ': {'main': 6, 'roman... |
| {'ADJ': {'third': 2, 'cruc... |
| {'ADJ': {'main': 2, 'ethni... |
| {'ADJ': {'first': 1, 'whic... |
| {'ADJ': {'aseptic': 1, 'ri... |
+-------------------------------+
[72269 rows x 1 columns]

Note that this API requires spaCy to be installed.

Sentence Splitting

For an SArray of strings, where each row is assumed to be a natural English language document, the sentence splitter splits by sentence and outputs a list of sentences. This aids in anlysis at the sentence level. For example, you may want a sentiment score for each sentence in a document. The following command accomplishes this for you:

sentences = graphlab.SFrame()
sentences['sent'] = graphlab.text_analytics.split_by_sentence(sf['X1'])
sentences
Columns:
  sent  list

Rows: 10

Data:
+-------------------------------+
|              sent             |
+-------------------------------+
| [alainconnes alain connes ... |
| [americannationalstandards... |
| [alberteinstein near the b... |
| [austriangerman as german ... |
| [arsenic arsenic is a meta... |
| [alps the alps alpen alpi ... |
| [alexiscarrel born in sain... |
| [adelaide adelaide is a co... |
| [artist an artist is a per... |
| [abdominalsurgery the thre... |
+-------------------------------+
[72269 rows x 1 columns]

Note that this API requires spaCy to be installed.