mirror of
https://github.com/gsi-upm/senpy
synced 2024-11-10 19:12:28 +00:00
1087692de2
* Add sklearn example * Fix test_case * Add SenpyClientUse docs a.k.a. The wise men edition
34 lines
631 B
Python
34 lines
631 B
Python
'''
|
|
Create a dummy dataset.
|
|
Messages with a happy emoticon are labelled positive
|
|
Messages with a sad emoticon are labelled negative
|
|
'''
|
|
import random
|
|
|
|
dataset = []
|
|
|
|
vocabulary = ['hello', 'world', 'senpy', 'cool', 'goodbye', 'random', 'text']
|
|
|
|
emojimap = {
|
|
1: [':)', ],
|
|
-1: [':(', ]
|
|
}
|
|
|
|
|
|
for tag, values in emojimap.items():
|
|
for i in range(1000):
|
|
msg = ''
|
|
for j in range(3):
|
|
msg += random.choice(vocabulary)
|
|
msg += " "
|
|
msg += random.choice(values)
|
|
dataset.append([msg, tag])
|
|
|
|
|
|
text = []
|
|
labels = []
|
|
|
|
for i in dataset:
|
|
text.append(i[0])
|
|
labels.append(i[1])
|