def test_can_delete(self):
s = Snapper()
record = {'target': 1, 'data': {'fieldStrength': 20}}
s.append(record)
s.append(record)
s.append(record)
uuidToDel = s.data[1]['uuid']
expect(len(s.data)).to_equal(3)
s.remove(uuidToDel)
expect(len(s.data)).to_equal(2)
def test_can_append(self):
s = Snapper()
record = {'target': 1, 'data': {'fieldStrength': 20}}
s.append(record)
s.append(record)
s.append(record)
s.save('temp.dataset')
s2 = Snapper()
s2.load('temp.dataset')
expect(s2.data[0].get('target')).to_equal(1)
expect(s2.data[2].get('target')).to_equal(1)
expect(s2.data[2].get('data').get('fieldStrength')).to_equal(20)
def test_can_delete(self):
s = Snapper()
record = {
'target': 1,
'data': {
'fieldStrength': 20
}
}
s.append(record)
s.append(record)
s.append(record)
uuidToDel = s.data[1]['uuid']
expect(len(s.data)).to_equal(3)
s.remove(uuidToDel)
expect(len(s.data)).to_equal(2)
def test_can_append(self):
s = Snapper()
record = {
'target': 1,
'data': {
'fieldStrength': 20
}
}
s.append(record)
s.append(record)
s.append(record)
s.save('temp.dataset')
s2 = Snapper()
s2.load('temp.dataset')
expect(s2.data[0].get('target')).to_equal(1)
expect(s2.data[2].get('target')).to_equal(1)
expect(s2.data[2].get('data').get('fieldStrength')).to_equal(20)
def test_can_train(self):
s = Snapper()
record = {'target': 1, 'data': {'fieldStrength': 20}}
record2 = {'target': 2, 'data': {'fieldStrength': 40, 'temp': 28.8}}
s.append(record)
s.append(record2)
s.append(record)
s.trainKNN(neighbors=3)
def test_can_export(self):
s = Snapper()
record = {'target': 1, 'data': {'fieldStrength': 20}}
record2 = {'target': 2, 'data': {'fieldStrength': 40, 'temp': 28.8}}
s.append(record)
s.append(record2)
s.append(record)
s.export('temp.dataset.exptd')
def test_can_train(self):
s = Snapper()
record = {
'target': 1,
'data': {
'fieldStrength': 20
}
}
record2 = {
'target': 2,
'data': {
'fieldStrength': 40,
'temp': 28.8
}
}
s.append(record)
s.append(record2)
s.append(record)
s.trainKNN(neighbors=3)
def test_can_export(self):
s = Snapper()
record = {
'target': 1,
'data': {
'fieldStrength': 20
}
}
record2 = {
'target': 2,
'data': {
'fieldStrength': 40,
'temp': 28.8
}
}
s.append(record)
s.append(record2)
s.append(record)
s.export('temp.dataset.exptd')
class SnapperWidget(Gtk.Window):
def __init__(self):
import os
print os.getcwd()
## File accessed from parent folder.
filename = "easyNav_snapper_ui/snapTemplate.glade"
self.builder = Gtk.Builder()
self.builder.add_from_file(filename)
self.builder.connect_signals(self)
## For Snapper stuff
self.snapper = Snapper()
self.saveFilepath = None
## For SerialDaemon stuff
sd = self.sd = SerialDaemon()
## For wifi stuff
self.WIFI_STABLIZATION_TIME = 4
self.networksLock = threading.Lock() # make wifi data atomic
swd = self.swd = SensorWifiDaemon(interval=self.WIFI_STABLIZATION_TIME, interface='wlan0')
## Attach data handlers
self._attachHandlers()
sd.configure()
sd.start()
swd.start()
## For collection start / stop button and actions
self._isCollecting = False
## for the store
self.store = Gtk.ListStore(int, str, str)
tree = self.builder.get_object('treeview3')
tree.set_model(self.store)
# For thresholds
self.WIFI_RSSI_THRESHOLD = 0
self.THRESHOLD_MIN_NETWORKS = 0
## for B field details. This is needed to prevent the delay observed on reading
## from text field.
self.bField = {
'x': 0,
'y': 0,
'z': 0,
'intensity': 0
}
renderer = Gtk.CellRendererText()
column = Gtk.TreeViewColumn("Target", renderer, text=0)
column.set_sort_column_id(0)
tree.append_column(column)
column2 = Gtk.TreeViewColumn("Data", renderer, text=1)
tree.append_column(column2)
## update the builder
self.builder.get_object('window1').show_all()
def clearTreeView(self):
""" Clears the tree view
"""
self.store.clear()
def populateTreeView(self):
""" Populates tree with store items. used for load.
"""
for item in self.snapper.data:
treeiter = self.store.append([
item.get('target'),
str(item.get('data')),
str(item.get('uuid')) ])
def on_treeview3_cursor_changed(self, args):
pass
def on_btnDelete_clicked(self, args):
tree = self.builder.get_object('treeview3')
model, itr = tree.get_selection().get_selected()
uuid = model[itr][2]
model.remove(itr)
self.snapper.remove(int(uuid))
def _updateBField(self, x,y,z,intensity):
self.builder.get_object('bField_x').set_text(str(x))
self.builder.get_object('bField_y').set_text(str(y))
self.builder.get_object('bField_z').set_text(str(z))
self.builder.get_object('bField_norm').set_text(str(intensity))
def _attachHandlers(self):
"""Event listeners
"""
## TODO: Does not trigger when GTK starts. Maybe try fork?
@smokesignal.on('onData')
def onDataHandler(x,y,z,intensity):
""" Event callback for serial data
"""
self.bField = {
'x': x,
'y': y,
'z': z,
'intensity': intensity
}
## TODO: Remove the display. might slow down system.
# GObject.idle_add(self._updateBField, x,y,z,intensity)
# logging.info('Serial Daemon: New Data!')
# print (x,y,z,intensity)
@smokesignal.on('onNetworkData')
def onNetworkDataHandler(networks):
"""Event callback for wifi data
"""
result = {}
for n in networks:
essid = str(n['Address'])
quality = int(n['Quality'])
# If less than threshold, do not include!
if (quality >= self.WIFI_RSSI_THRESHOLD):
result[essid] = quality
## Write data to atomic self.networks
self.networksLock.acquire()
self.networks = result
self.networksLock.release()
def buttonWriteNameToFile_clicked(self, widget):
print("File write code...")
def on_window1_destroy(self, *args):
""" Closes the main program
"""
self.sd.stop()
self.swd.stop()
Gtk.main_quit()
def on_imagemenuitem2_activate(self, args):
""" Load file
"""
print 'selected menu'
chooser_dialog = Gtk.FileChooserDialog(title='Load file..'
, action=Gtk.FileChooserAction.OPEN
, buttons=["Open", Gtk.ResponseType.OK, "Cancel", Gtk.ResponseType.CANCEL])
pyFilter = Gtk.FileFilter()
pyFilter.add_mime_type('text/x-python')
pyFilter.add_pattern('*.py')
pyFilter.set_name('Python files')
response = chooser_dialog.run()
if response == Gtk.ResponseType.OK:
filename = chooser_dialog.get_filename()
# Load information into snapper
self.snapper.load(filename)
# populate treeview
self.clearTreeView()
self.populateTreeView()
self.builder.get_object('statusbar1').push(0, 'File %s loaded.' % filename)
elif response == Gtk.ResponseType.CANCEL:
pass
chooser_dialog.destroy()
def on_imagemenuitem4_activate(self, args):
""" Save as..
"""
print 'selected menu'
chooser_dialog = Gtk.FileChooserDialog(title='Load file..'
, action=Gtk.FileChooserAction.SAVE
, buttons=["Save as..", Gtk.ResponseType.OK, "Cancel", Gtk.ResponseType.CANCEL])
Gtk.FileChooser.set_do_overwrite_confirmation(chooser_dialog, True)
pyFilter = Gtk.FileFilter()
pyFilter.add_mime_type('text/x-python')
pyFilter.add_pattern('*.py')
pyFilter.set_name('Python files')
response = chooser_dialog.run()
if response == Gtk.ResponseType.OK:
filename = chooser_dialog.get_filename()
# Save information from snapper
self.snapper.save(filename)
self.builder.get_object('statusbar1').push(0, 'File %s loaded.' % filename)
elif response == Gtk.ResponseType.CANCEL:
pass
chooser_dialog.destroy()
def on_btnExport_activate(self, args):
""" Export to SVM Light format
"""
print 'selected menu'
chooser_dialog = Gtk.FileChooserDialog(title='Load file..'
, action=Gtk.FileChooserAction.SAVE
, buttons=["Export to..", Gtk.ResponseType.OK, "Cancel", Gtk.ResponseType.CANCEL])
Gtk.FileChooser.set_do_overwrite_confirmation(chooser_dialog, True)
pyFilter = Gtk.FileFilter()
pyFilter.add_mime_type('text/x-python')
pyFilter.add_pattern('*.py')
pyFilter.set_name('Python files')
response = chooser_dialog.run()
if response == Gtk.ResponseType.OK:
filename = chooser_dialog.get_filename()
# Save information from snapper
self.snapper.export(filename)
self.builder.get_object('statusbar1').push(0, 'SVMLight File %s exported.' % filename)
elif response == Gtk.ResponseType.CANCEL:
pass
chooser_dialog.destroy()
def on_btnUpdateThresholds_clicked(self, args):
"""Button handler to update thresholds
"""
wifi_threshold = int(self.builder.get_object('fieldThresholdRssi').get_text())
minNetwork_threshold = int(self.builder.get_object('fieldMinNetwork').get_text())
self.WIFI_RSSI_THRESHOLD = wifi_threshold
self.THRESHOLD_MIN_NETWORKS = minNetwork_threshold
logging.debug("WIFI_RSSI_THRESHOLD: %s, THRESHOLD_MIN_NETWORKS: %s" % (self.WIFI_RSSI_THRESHOLD, self.THRESHOLD_MIN_NETWORKS))
self.builder.get_object('statusbar1').push(0, 'Updated thresholds.')
def on_btnWifiUpdate_clicked(self, args):
""" Button to update wifi interval and port
"""
self.WIFI_STABLIZATION_TIME = wifiInterval = float(self.builder.get_object('wifiInterval').get_text())
wifiPort = str(self.builder.get_object('wifiPort').get_text())
self.swd.updateConfig(wifiInterval, wifiPort)
self.builder.get_object('statusbar1').push(
0, 'Updated WIFI: port=%s interval=%s' % (wifiPort, wifiInterval))
def on_btnAdd_clicked(self, args):
""" Add a new Entry
"""
## Run tick thread
def runThread():
_refTime = time.time()
while(self._isCollecting):
# If it is not the right time,
# do not execute
if ((time.time() - _refTime) < self.WIFI_STABLIZATION_TIME):
time.sleep(0.05)
else:
## Reset reference time
_refTime = time.time()
self.addNewEntry()
print 'tick..'
print 'stopped thread.'
self._isCollecting = not self._isCollecting
if (self._isCollecting):
self.builder.get_object('btnAdd').set_label('Stop')
## Start data collection
self._threadCollect = threading.Thread(target=runThread)
self._threadCollect.start()
else:
self.builder.get_object('btnAdd').set_label('Start')
## Stop data collection
self._threadCollect.join()
def addNewEntry(self):
""" Helper function to add new entry.
"""
# bFieldX = float(self.builder.get_object('bField_x').get_text())
# bFieldY = float(self.builder.get_object('bField_y').get_text())
# bFieldZ = float(self.builder.get_object('bField_z').get_text())
# bFieldMagnitude = float(self.builder.get_object('bField_norm').get_text())
## Use the B field var instead of text boxes to reduce lag
bFieldX = self.bField['x']
bFieldY = self.bField['y']
bFieldZ = self.bField['z']
bFieldMagnitude = self.bField['intensity']
locId = int(self.builder.get_object('locId').get_text())
## Get B Field Data
bFieldData = {
'bField': bFieldMagnitude,
'bFieldX': self.bField['x'],
'bFieldY': self.bField['y']
# 'bFieldz': self.bField['z']
}
## Get network data
## Get data from atomic self.networks
self.networksLock.acquire()
networkData = self.networks
self.networksLock.release()
# Do not add data entry if number of networks
# is too small. This is needed to avoid diluting
# feature set.
if ( len(networkData) < self.THRESHOLD_MIN_NETWORKS):
return
# Do not add if no network data
if ( len(networkData) == 0):
return
## Get combined data
## TODO: Uncomment this line for B field inclusion
combinedData = dict(bFieldData.items() + networkData.items())
## Uncomment this line to only get network data
# combinedData = networkData
item = {
'target': locId,
'data': combinedData
}
itemDict = self.snapper.append(item)
self.builder.get_object('statusbar1').push(0, 'Added [%s]@%s: B Field=%s' %
(len(self.snapper.data),
locId,
bFieldMagnitude))
# show in table
idx = len(self.snapper.data) - 1
self.store.append([item['target'], str(item['data']), str(itemDict['uuid'])])
def on_btnTrain_clicked(self, args):
""" Train button
"""
choice = 'KNN'
## Get selection
if (self.builder.get_object('radioKNN').get_active()):
choice = 'KNN'
else:
choice = 'SVM'
## Number of clusters to use for feature reduction
n_clusters = int(self.builder.get_object('entryClusters').get_text())
if (choice == 'KNN'):
# Train using KNN -----
neighbors = int(self.builder.get_object('entryNeighbors').get_text())
self.snapper.trainKNN(n_clusters, neighbors)
self.builder.get_object('statusbar1').push(
0, 'Trained KNN model with N=%s Clusters=%s' % (neighbors, n_clusters))
else:
# Train using SVM -----
C=int(self.builder.get_object('svmC').get_text())
cache_size=int(self.builder.get_object('svmCache').get_text())
class_weight=None
coef0=0.0
degree=int(self.builder.get_object('svmDegree').get_text())
gamma=float(self.builder.get_object('svmGamma').get_text())
kernel=str(self.builder.get_object('svmKernel').get_text())
max_iter=-1
probability=False
random_state=None
shrinking=True
tol=float(self.builder.get_object('svmTol').get_text())
verbose=False
self.snapper.trainSVM(
n_clusters=n_clusters,
C=C,
cache_size=cache_size,
class_weight=class_weight,
coef0=coef0,
degree=degree,
gamma=gamma,
kernel=kernel,
max_iter=max_iter,
probability=probability,
random_state=random_state,
shrinking=shrinking,
tol=tol,
verbose=verbose)
self.builder.get_object('statusbar1').push(
0, 'Trained SVM model with C=%s cache=%s gamma=%s ker=%s tol=%s Clusters=%s' %
(C, cache_size, gamma, kernel, tol, n_clusters ))
def on_btnPredict_clicked(self, args):
""" Predict button
"""
## Populate test item vector with zeroes first
item = {}
keys = self.snapper.getKeys()
for k in keys:
item[k] = 0
## Get B Field Data
# bFieldMagnitude = float(self.builder.get_object('bField_norm').get_text())
bFieldMagnitude = self.bField['intensity']
bFieldData = {
'bField': bFieldMagnitude,
'bFieldX': self.bField['x'],
'bFieldY': self.bField['y']
# 'bFieldZ': self.bField['z']
}
## Get network data
## Get data from atomic self.networks
self.networksLock.acquire()
networkData = self.networks
self.networksLock.release()
# print networkData
## Get combined data
## TODO: Uncomment this line for B field inclusion
combinedData = dict(bFieldData.items() + networkData.items())
## Uncomment line below to get only network data.
combinedData = networkData
## Merge combined data
for k in keys:
if k in combinedData:
item[k] = combinedData[k]
prediction = self.snapper.predict(item)
logging.info('Predicted: %s' % item)
logging.info('Prediction: %s' % prediction)
self.builder.get_object('statusbar1').push(0, 'The prediction is: %s' % prediction)
def on_btnSerialConnect_clicked(self, args):
"""Connect Serial button handler
"""
sensorType = 'phone'
port = str(self.builder.get_object('fieldSerialPort').get_text())
if (self.builder.get_object('radioSensorPhone').get_active()):
sensorType = 'phone'
else:
sensorType = 'shoe'
self.sd.stop()
self.sd.configure(port, sensorType)
self.sd.start()
self.builder.get_object('statusbar1').push(0, 'Serial port=%s type=%s started.' % (port, sensorType))
def on_btnSerialDisconnect_clicked(self, args):
self.sd.stop()
self.builder.get_object('statusbar1').push(0, 'Serial port stopped.')
def test_svm_can_work(self):
s = Snapper()
record = {'target': 1, 'data': {'fieldStrength': 20, 'temp': 28.8}}
record2 = {'target': 2, 'data': {'fieldStrength': 40, 'temp': 28.8}}
s.append(record)
s.append(record2)
s.append(record2)
s.append(record2)
s.append(record2)
s.append(record2)
s.append(record2)
s.append(record2)
s.append(record2)
s.append(record2)
s.append(record2)
s.append(record2)
s.append(record)
s.trainSVM()
prediction = s.predict({'fieldStrength': 25, 'temp': 28.8})
print 'prediction: ', prediction, '-------'
class SnapperWidget(Gtk.Window):
def __init__(self):
import os
print os.getcwd()
## File accessed from parent folder.
filename = "easyNav_snapper_ui/snapTemplate.glade"
self.builder = Gtk.Builder()
self.builder.add_from_file(filename)
self.builder.connect_signals(self)
## For Snapper stuff
self.snapper = Snapper()
self.saveFilepath = None
## For SerialDaemon stuff
sd = self.sd = SerialDaemon()
## For wifi stuff
self.WIFI_STABLIZATION_TIME = 4
self.networksLock = threading.Lock() # make wifi data atomic
swd = self.swd = SensorWifiDaemon(interval=self.WIFI_STABLIZATION_TIME,
interface='wlan0')
## Attach data handlers
self._attachHandlers()
sd.configure()
sd.start()
swd.start()
## For collection start / stop button and actions
self._isCollecting = False
## for the store
self.store = Gtk.ListStore(int, str, str)
tree = self.builder.get_object('treeview3')
tree.set_model(self.store)
# For thresholds
self.WIFI_RSSI_THRESHOLD = 0
self.THRESHOLD_MIN_NETWORKS = 0
## for B field details. This is needed to prevent the delay observed on reading
## from text field.
self.bField = {'x': 0, 'y': 0, 'z': 0, 'intensity': 0}
renderer = Gtk.CellRendererText()
column = Gtk.TreeViewColumn("Target", renderer, text=0)
column.set_sort_column_id(0)
tree.append_column(column)
column2 = Gtk.TreeViewColumn("Data", renderer, text=1)
tree.append_column(column2)
## update the builder
self.builder.get_object('window1').show_all()
def clearTreeView(self):
""" Clears the tree view
"""
self.store.clear()
def populateTreeView(self):
""" Populates tree with store items. used for load.
"""
for item in self.snapper.data:
treeiter = self.store.append([
item.get('target'),
str(item.get('data')),
str(item.get('uuid'))
])
def on_treeview3_cursor_changed(self, args):
pass
def on_btnDelete_clicked(self, args):
tree = self.builder.get_object('treeview3')
model, itr = tree.get_selection().get_selected()
uuid = model[itr][2]
model.remove(itr)
self.snapper.remove(int(uuid))
def _updateBField(self, x, y, z, intensity):
self.builder.get_object('bField_x').set_text(str(x))
self.builder.get_object('bField_y').set_text(str(y))
self.builder.get_object('bField_z').set_text(str(z))
self.builder.get_object('bField_norm').set_text(str(intensity))
def _attachHandlers(self):
"""Event listeners
"""
## TODO: Does not trigger when GTK starts. Maybe try fork?
@smokesignal.on('onData')
def onDataHandler(x, y, z, intensity):
""" Event callback for serial data
"""
self.bField = {'x': x, 'y': y, 'z': z, 'intensity': intensity}
## TODO: Remove the display. might slow down system.
# GObject.idle_add(self._updateBField, x,y,z,intensity)
# logging.info('Serial Daemon: New Data!')
# print (x,y,z,intensity)
@smokesignal.on('onNetworkData')
def onNetworkDataHandler(networks):
"""Event callback for wifi data
"""
result = {}
for n in networks:
essid = str(n['Address'])
quality = int(n['Quality'])
# If less than threshold, do not include!
if (quality >= self.WIFI_RSSI_THRESHOLD):
result[essid] = quality
## Write data to atomic self.networks
self.networksLock.acquire()
self.networks = result
self.networksLock.release()
def buttonWriteNameToFile_clicked(self, widget):
print("File write code...")
def on_window1_destroy(self, *args):
""" Closes the main program
"""
self.sd.stop()
self.swd.stop()
Gtk.main_quit()
def on_imagemenuitem2_activate(self, args):
""" Load file
"""
print 'selected menu'
chooser_dialog = Gtk.FileChooserDialog(
title='Load file..',
action=Gtk.FileChooserAction.OPEN,
buttons=[
"Open", Gtk.ResponseType.OK, "Cancel", Gtk.ResponseType.CANCEL
])
pyFilter = Gtk.FileFilter()
pyFilter.add_mime_type('text/x-python')
pyFilter.add_pattern('*.py')
pyFilter.set_name('Python files')
response = chooser_dialog.run()
if response == Gtk.ResponseType.OK:
filename = chooser_dialog.get_filename()
# Load information into snapper
self.snapper.load(filename)
# populate treeview
self.clearTreeView()
self.populateTreeView()
self.builder.get_object('statusbar1').push(
0, 'File %s loaded.' % filename)
elif response == Gtk.ResponseType.CANCEL:
pass
chooser_dialog.destroy()
def on_imagemenuitem4_activate(self, args):
""" Save as..
"""
print 'selected menu'
chooser_dialog = Gtk.FileChooserDialog(
title='Load file..',
action=Gtk.FileChooserAction.SAVE,
buttons=[
"Save as..", Gtk.ResponseType.OK, "Cancel",
Gtk.ResponseType.CANCEL
])
Gtk.FileChooser.set_do_overwrite_confirmation(chooser_dialog, True)
pyFilter = Gtk.FileFilter()
pyFilter.add_mime_type('text/x-python')
pyFilter.add_pattern('*.py')
pyFilter.set_name('Python files')
response = chooser_dialog.run()
if response == Gtk.ResponseType.OK:
filename = chooser_dialog.get_filename()
# Save information from snapper
self.snapper.save(filename)
self.builder.get_object('statusbar1').push(
0, 'File %s loaded.' % filename)
elif response == Gtk.ResponseType.CANCEL:
pass
chooser_dialog.destroy()
def on_btnExport_activate(self, args):
""" Export to SVM Light format
"""
print 'selected menu'
chooser_dialog = Gtk.FileChooserDialog(
title='Load file..',
action=Gtk.FileChooserAction.SAVE,
buttons=[
"Export to..", Gtk.ResponseType.OK, "Cancel",
Gtk.ResponseType.CANCEL
])
Gtk.FileChooser.set_do_overwrite_confirmation(chooser_dialog, True)
pyFilter = Gtk.FileFilter()
pyFilter.add_mime_type('text/x-python')
pyFilter.add_pattern('*.py')
pyFilter.set_name('Python files')
response = chooser_dialog.run()
if response == Gtk.ResponseType.OK:
filename = chooser_dialog.get_filename()
# Save information from snapper
self.snapper.export(filename)
self.builder.get_object('statusbar1').push(
0, 'SVMLight File %s exported.' % filename)
elif response == Gtk.ResponseType.CANCEL:
pass
chooser_dialog.destroy()
def on_btnUpdateThresholds_clicked(self, args):
"""Button handler to update thresholds
"""
wifi_threshold = int(
self.builder.get_object('fieldThresholdRssi').get_text())
minNetwork_threshold = int(
self.builder.get_object('fieldMinNetwork').get_text())
self.WIFI_RSSI_THRESHOLD = wifi_threshold
self.THRESHOLD_MIN_NETWORKS = minNetwork_threshold
logging.debug("WIFI_RSSI_THRESHOLD: %s, THRESHOLD_MIN_NETWORKS: %s" %
(self.WIFI_RSSI_THRESHOLD, self.THRESHOLD_MIN_NETWORKS))
self.builder.get_object('statusbar1').push(0, 'Updated thresholds.')
def on_btnWifiUpdate_clicked(self, args):
""" Button to update wifi interval and port
"""
self.WIFI_STABLIZATION_TIME = wifiInterval = float(
self.builder.get_object('wifiInterval').get_text())
wifiPort = str(self.builder.get_object('wifiPort').get_text())
self.swd.updateConfig(wifiInterval, wifiPort)
self.builder.get_object('statusbar1').push(
0, 'Updated WIFI: port=%s interval=%s' % (wifiPort, wifiInterval))
def on_btnAdd_clicked(self, args):
""" Add a new Entry
"""
## Run tick thread
def runThread():
_refTime = time.time()
while (self._isCollecting):
# If it is not the right time,
# do not execute
if ((time.time() - _refTime) < self.WIFI_STABLIZATION_TIME):
time.sleep(0.05)
else:
## Reset reference time
_refTime = time.time()
self.addNewEntry()
print 'tick..'
print 'stopped thread.'
self._isCollecting = not self._isCollecting
if (self._isCollecting):
self.builder.get_object('btnAdd').set_label('Stop')
## Start data collection
self._threadCollect = threading.Thread(target=runThread)
self._threadCollect.start()
else:
self.builder.get_object('btnAdd').set_label('Start')
## Stop data collection
self._threadCollect.join()
def addNewEntry(self):
""" Helper function to add new entry.
"""
# bFieldX = float(self.builder.get_object('bField_x').get_text())
# bFieldY = float(self.builder.get_object('bField_y').get_text())
# bFieldZ = float(self.builder.get_object('bField_z').get_text())
# bFieldMagnitude = float(self.builder.get_object('bField_norm').get_text())
## Use the B field var instead of text boxes to reduce lag
bFieldX = self.bField['x']
bFieldY = self.bField['y']
bFieldZ = self.bField['z']
bFieldMagnitude = self.bField['intensity']
locId = int(self.builder.get_object('locId').get_text())
## Get B Field Data
bFieldData = {
'bField': bFieldMagnitude,
'bFieldX': self.bField['x'],
'bFieldY': self.bField['y']
# 'bFieldz': self.bField['z']
}
## Get network data
## Get data from atomic self.networks
self.networksLock.acquire()
networkData = self.networks
self.networksLock.release()
# Do not add data entry if number of networks
# is too small. This is needed to avoid diluting
# feature set.
if (len(networkData) < self.THRESHOLD_MIN_NETWORKS):
return
# Do not add if no network data
if (len(networkData) == 0):
return
## Get combined data
## TODO: Uncomment this line for B field inclusion
combinedData = dict(bFieldData.items() + networkData.items())
## Uncomment this line to only get network data
# combinedData = networkData
item = {'target': locId, 'data': combinedData}
itemDict = self.snapper.append(item)
self.builder.get_object('statusbar1').push(
0, 'Added [%s]@%s: B Field=%s' %
(len(self.snapper.data), locId, bFieldMagnitude))
# show in table
idx = len(self.snapper.data) - 1
self.store.append(
[item['target'],
str(item['data']),
str(itemDict['uuid'])])
def on_btnTrain_clicked(self, args):
""" Train button
"""
choice = 'KNN'
## Get selection
if (self.builder.get_object('radioKNN').get_active()):
choice = 'KNN'
else:
choice = 'SVM'
## Number of clusters to use for feature reduction
n_clusters = int(self.builder.get_object('entryClusters').get_text())
if (choice == 'KNN'):
# Train using KNN -----
neighbors = int(
self.builder.get_object('entryNeighbors').get_text())
self.snapper.trainKNN(n_clusters, neighbors)
self.builder.get_object('statusbar1').push(
0, 'Trained KNN model with N=%s Clusters=%s' %
(neighbors, n_clusters))
else:
# Train using SVM -----
C = int(self.builder.get_object('svmC').get_text())
cache_size = int(self.builder.get_object('svmCache').get_text())
class_weight = None
coef0 = 0.0
degree = int(self.builder.get_object('svmDegree').get_text())
gamma = float(self.builder.get_object('svmGamma').get_text())
kernel = str(self.builder.get_object('svmKernel').get_text())
max_iter = -1
probability = False
random_state = None
shrinking = True
tol = float(self.builder.get_object('svmTol').get_text())
verbose = False
self.snapper.trainSVM(n_clusters=n_clusters,
C=C,
cache_size=cache_size,
class_weight=class_weight,
coef0=coef0,
degree=degree,
gamma=gamma,
kernel=kernel,
max_iter=max_iter,
probability=probability,
random_state=random_state,
shrinking=shrinking,
tol=tol,
verbose=verbose)
self.builder.get_object('statusbar1').push(
0,
'Trained SVM model with C=%s cache=%s gamma=%s ker=%s tol=%s Clusters=%s'
% (C, cache_size, gamma, kernel, tol, n_clusters))
def on_btnPredict_clicked(self, args):
""" Predict button
"""
## Populate test item vector with zeroes first
item = {}
keys = self.snapper.getKeys()
for k in keys:
item[k] = 0
## Get B Field Data
# bFieldMagnitude = float(self.builder.get_object('bField_norm').get_text())
bFieldMagnitude = self.bField['intensity']
bFieldData = {
'bField': bFieldMagnitude,
'bFieldX': self.bField['x'],
'bFieldY': self.bField['y']
# 'bFieldZ': self.bField['z']
}
## Get network data
## Get data from atomic self.networks
self.networksLock.acquire()
networkData = self.networks
self.networksLock.release()
# print networkData
## Get combined data
## TODO: Uncomment this line for B field inclusion
combinedData = dict(bFieldData.items() + networkData.items())
## Uncomment line below to get only network data.
combinedData = networkData
## Merge combined data
for k in keys:
if k in combinedData:
item[k] = combinedData[k]
prediction = self.snapper.predict(item)
logging.info('Predicted: %s' % item)
logging.info('Prediction: %s' % prediction)
self.builder.get_object('statusbar1').push(
0, 'The prediction is: %s' % prediction)
def on_btnSerialConnect_clicked(self, args):
"""Connect Serial button handler
"""
sensorType = 'phone'
port = str(self.builder.get_object('fieldSerialPort').get_text())
if (self.builder.get_object('radioSensorPhone').get_active()):
sensorType = 'phone'
else:
sensorType = 'shoe'
self.sd.stop()
self.sd.configure(port, sensorType)
self.sd.start()
self.builder.get_object('statusbar1').push(
0, 'Serial port=%s type=%s started.' % (port, sensorType))
def on_btnSerialDisconnect_clicked(self, args):
self.sd.stop()
self.builder.get_object('statusbar1').push(0, 'Serial port stopped.')
def test_svm_can_work(self):
s = Snapper()
record = {
'target': 1,
'data': {
'fieldStrength': 20,
'temp': 28.8
}
}
record2 = {
'target': 2,
'data': {
'fieldStrength': 40,
'temp': 28.8
}
}
s.append(record)
s.append(record2)
s.append(record2)
s.append(record2)
s.append(record2)
s.append(record2)
s.append(record2)
s.append(record2)
s.append(record2)
s.append(record2)
s.append(record2)
s.append(record2)
s.append(record)
s.trainSVM()
prediction = s.predict({
'fieldStrength' : 25,
'temp': 28.8
})
print 'prediction: ', prediction, '-------'