Python Text Analysis with TextBlob — Reading lines from The Office TV Show
$begingroup$
I have a Jupyter notebook that allows a user to type in a phrase and the program will 'guess' which Office character it resembles.
Here is a link to the github repo
I have the following littered throughout my code:
the_office_raw_script['polarity'] = the_office_raw_script['line_text'].apply(lambda x: TextBlob(x).sentiment.polarity)
the_office_raw_script['scores'] = the_office_raw_script['line_text'].apply(lambda x: analyser.polarity_scores(x))
and the following, which is the prediction part:
from sklearn.model_selection import train_test_split
from sklearn.feature_extraction.text import CountVectorizer
from sklearn.feature_extraction.text import TfidfTransformer
from sklearn.naive_bayes import MultinomialNB
X_train, X_test, y_train, y_test = train_test_split(df_upsampled['line_text'], df_upsampled['speaker'], random_state = 0)
count_vect = CountVectorizer()
X_train_counts = count_vect.fit_transform(X_train)
tfidf_transformer = TfidfTransformer()
X_train_tfidf = tfidf_transformer.fit_transform(X_train_counts)
clf = MultinomialNB().fit(X_train_tfidf, y_train)
I have some examples of output, which is AWESOME! (so happy i got this far)
print(clf.predict(count_vect.transform(["hard working beet farmer"])))
the above prints dwight
print(clf.predict(count_vect.transform(["that's what she said"])))
the above prints pam
which is totally wrong because michael says 'that's what she said' over 15 times throughout the series.
Two Questions
How would I go above making my script into a function, would that be worth it?
How do I somewhat ~hardcode~ some inputs? If someone types in 'that's what she said' or a variation thereof I want it to always print michael
?
python
New contributor
$endgroup$
add a comment |
$begingroup$
I have a Jupyter notebook that allows a user to type in a phrase and the program will 'guess' which Office character it resembles.
Here is a link to the github repo
I have the following littered throughout my code:
the_office_raw_script['polarity'] = the_office_raw_script['line_text'].apply(lambda x: TextBlob(x).sentiment.polarity)
the_office_raw_script['scores'] = the_office_raw_script['line_text'].apply(lambda x: analyser.polarity_scores(x))
and the following, which is the prediction part:
from sklearn.model_selection import train_test_split
from sklearn.feature_extraction.text import CountVectorizer
from sklearn.feature_extraction.text import TfidfTransformer
from sklearn.naive_bayes import MultinomialNB
X_train, X_test, y_train, y_test = train_test_split(df_upsampled['line_text'], df_upsampled['speaker'], random_state = 0)
count_vect = CountVectorizer()
X_train_counts = count_vect.fit_transform(X_train)
tfidf_transformer = TfidfTransformer()
X_train_tfidf = tfidf_transformer.fit_transform(X_train_counts)
clf = MultinomialNB().fit(X_train_tfidf, y_train)
I have some examples of output, which is AWESOME! (so happy i got this far)
print(clf.predict(count_vect.transform(["hard working beet farmer"])))
the above prints dwight
print(clf.predict(count_vect.transform(["that's what she said"])))
the above prints pam
which is totally wrong because michael says 'that's what she said' over 15 times throughout the series.
Two Questions
How would I go above making my script into a function, would that be worth it?
How do I somewhat ~hardcode~ some inputs? If someone types in 'that's what she said' or a variation thereof I want it to always print michael
?
python
New contributor
$endgroup$
add a comment |
$begingroup$
I have a Jupyter notebook that allows a user to type in a phrase and the program will 'guess' which Office character it resembles.
Here is a link to the github repo
I have the following littered throughout my code:
the_office_raw_script['polarity'] = the_office_raw_script['line_text'].apply(lambda x: TextBlob(x).sentiment.polarity)
the_office_raw_script['scores'] = the_office_raw_script['line_text'].apply(lambda x: analyser.polarity_scores(x))
and the following, which is the prediction part:
from sklearn.model_selection import train_test_split
from sklearn.feature_extraction.text import CountVectorizer
from sklearn.feature_extraction.text import TfidfTransformer
from sklearn.naive_bayes import MultinomialNB
X_train, X_test, y_train, y_test = train_test_split(df_upsampled['line_text'], df_upsampled['speaker'], random_state = 0)
count_vect = CountVectorizer()
X_train_counts = count_vect.fit_transform(X_train)
tfidf_transformer = TfidfTransformer()
X_train_tfidf = tfidf_transformer.fit_transform(X_train_counts)
clf = MultinomialNB().fit(X_train_tfidf, y_train)
I have some examples of output, which is AWESOME! (so happy i got this far)
print(clf.predict(count_vect.transform(["hard working beet farmer"])))
the above prints dwight
print(clf.predict(count_vect.transform(["that's what she said"])))
the above prints pam
which is totally wrong because michael says 'that's what she said' over 15 times throughout the series.
Two Questions
How would I go above making my script into a function, would that be worth it?
How do I somewhat ~hardcode~ some inputs? If someone types in 'that's what she said' or a variation thereof I want it to always print michael
?
python
New contributor
$endgroup$
I have a Jupyter notebook that allows a user to type in a phrase and the program will 'guess' which Office character it resembles.
Here is a link to the github repo
I have the following littered throughout my code:
the_office_raw_script['polarity'] = the_office_raw_script['line_text'].apply(lambda x: TextBlob(x).sentiment.polarity)
the_office_raw_script['scores'] = the_office_raw_script['line_text'].apply(lambda x: analyser.polarity_scores(x))
and the following, which is the prediction part:
from sklearn.model_selection import train_test_split
from sklearn.feature_extraction.text import CountVectorizer
from sklearn.feature_extraction.text import TfidfTransformer
from sklearn.naive_bayes import MultinomialNB
X_train, X_test, y_train, y_test = train_test_split(df_upsampled['line_text'], df_upsampled['speaker'], random_state = 0)
count_vect = CountVectorizer()
X_train_counts = count_vect.fit_transform(X_train)
tfidf_transformer = TfidfTransformer()
X_train_tfidf = tfidf_transformer.fit_transform(X_train_counts)
clf = MultinomialNB().fit(X_train_tfidf, y_train)
I have some examples of output, which is AWESOME! (so happy i got this far)
print(clf.predict(count_vect.transform(["hard working beet farmer"])))
the above prints dwight
print(clf.predict(count_vect.transform(["that's what she said"])))
the above prints pam
which is totally wrong because michael says 'that's what she said' over 15 times throughout the series.
Two Questions
How would I go above making my script into a function, would that be worth it?
How do I somewhat ~hardcode~ some inputs? If someone types in 'that's what she said' or a variation thereof I want it to always print michael
?
python
python
New contributor
New contributor
New contributor
asked 6 hours ago
John FrielJohn Friel
1061
1061
New contributor
New contributor
add a comment |
add a comment |
0
active
oldest
votes
Your Answer
StackExchange.ifUsing("editor", function () {
return StackExchange.using("mathjaxEditing", function () {
StackExchange.MarkdownEditor.creationCallbacks.add(function (editor, postfix) {
StackExchange.mathjaxEditing.prepareWmdForMathJax(editor, postfix, [["\$", "\$"]]);
});
});
}, "mathjax-editing");
StackExchange.ifUsing("editor", function () {
StackExchange.using("externalEditor", function () {
StackExchange.using("snippets", function () {
StackExchange.snippets.init();
});
});
}, "code-snippets");
StackExchange.ready(function() {
var channelOptions = {
tags: "".split(" "),
id: "196"
};
initTagRenderer("".split(" "), "".split(" "), channelOptions);
StackExchange.using("externalEditor", function() {
// Have to fire editor after snippets, if snippets enabled
if (StackExchange.settings.snippets.snippetsEnabled) {
StackExchange.using("snippets", function() {
createEditor();
});
}
else {
createEditor();
}
});
function createEditor() {
StackExchange.prepareEditor({
heartbeatType: 'answer',
autoActivateHeartbeat: false,
convertImagesToLinks: false,
noModals: true,
showLowRepImageUploadWarning: true,
reputationToPostImages: null,
bindNavPrevention: true,
postfix: "",
imageUploader: {
brandingHtml: "Powered by u003ca class="icon-imgur-white" href="https://imgur.com/"u003eu003c/au003e",
contentPolicyHtml: "User contributions licensed under u003ca href="https://creativecommons.org/licenses/by-sa/3.0/"u003ecc by-sa 3.0 with attribution requiredu003c/au003e u003ca href="https://stackoverflow.com/legal/content-policy"u003e(content policy)u003c/au003e",
allowUrls: true
},
onDemand: true,
discardSelector: ".discard-answer"
,immediatelyShowMarkdownHelp:true
});
}
});
John Friel is a new contributor. Be nice, and check out our Code of Conduct.
Sign up or log in
StackExchange.ready(function () {
StackExchange.helpers.onClickDraftSave('#login-link');
});
Sign up using Google
Sign up using Facebook
Sign up using Email and Password
Post as a guest
Required, but never shown
StackExchange.ready(
function () {
StackExchange.openid.initPostLogin('.new-post-login', 'https%3a%2f%2fcodereview.stackexchange.com%2fquestions%2f213416%2fpython-text-analysis-with-textblob-reading-lines-from-the-office-tv-show%23new-answer', 'question_page');
}
);
Post as a guest
Required, but never shown
0
active
oldest
votes
0
active
oldest
votes
active
oldest
votes
active
oldest
votes
John Friel is a new contributor. Be nice, and check out our Code of Conduct.
John Friel is a new contributor. Be nice, and check out our Code of Conduct.
John Friel is a new contributor. Be nice, and check out our Code of Conduct.
John Friel is a new contributor. Be nice, and check out our Code of Conduct.
Thanks for contributing an answer to Code Review Stack Exchange!
- Please be sure to answer the question. Provide details and share your research!
But avoid …
- Asking for help, clarification, or responding to other answers.
- Making statements based on opinion; back them up with references or personal experience.
Use MathJax to format equations. MathJax reference.
To learn more, see our tips on writing great answers.
Sign up or log in
StackExchange.ready(function () {
StackExchange.helpers.onClickDraftSave('#login-link');
});
Sign up using Google
Sign up using Facebook
Sign up using Email and Password
Post as a guest
Required, but never shown
StackExchange.ready(
function () {
StackExchange.openid.initPostLogin('.new-post-login', 'https%3a%2f%2fcodereview.stackexchange.com%2fquestions%2f213416%2fpython-text-analysis-with-textblob-reading-lines-from-the-office-tv-show%23new-answer', 'question_page');
}
);
Post as a guest
Required, but never shown
Sign up or log in
StackExchange.ready(function () {
StackExchange.helpers.onClickDraftSave('#login-link');
});
Sign up using Google
Sign up using Facebook
Sign up using Email and Password
Post as a guest
Required, but never shown
Sign up or log in
StackExchange.ready(function () {
StackExchange.helpers.onClickDraftSave('#login-link');
});
Sign up using Google
Sign up using Facebook
Sign up using Email and Password
Post as a guest
Required, but never shown
Sign up or log in
StackExchange.ready(function () {
StackExchange.helpers.onClickDraftSave('#login-link');
});
Sign up using Google
Sign up using Facebook
Sign up using Email and Password
Sign up using Google
Sign up using Facebook
Sign up using Email and Password
Post as a guest
Required, but never shown
Required, but never shown
Required, but never shown
Required, but never shown
Required, but never shown
Required, but never shown
Required, but never shown
Required, but never shown
Required, but never shown