tornado_scikit_learn.py

#!/usr/bin/python
'''Starts and runs the scikit learn server'''

# For this to run properly, MongoDB must be running
#    Navigate to where mongo db is installed and run
#    something like $./mongod --dbpath "../data/db"
#    might need to use sudo (yikes!)

# database imports
from pymongo import MongoClient
from pymongo.errors import ServerSelectionTimeoutError


# tornado imports
import tornado.web
from tornado.web import HTTPError
from tornado.httpserver import HTTPServer
from tornado.ioloop import IOLoop
from tornado.options import define, options

# custom imports
from basehandler import BaseHandler
import sklearnhandlers as skh

# Setup information for tornado class
define("port", default=8000, help="run on the given port", type=int)

# Utility to be used when creating the Tornado server
# Contains the handlers and the database connection
class Application(tornado.web.Application):
    def __init__(self):
        '''Store necessary handlers,
           connect to database
        '''

        handlers = [(r"/[/]?", BaseHandler),
                    (r"/Handlers[/]?",        skh.PrintHandlers),
                    (r"/GetClasses[/]?",      skh.GetClasses),
                    (r"/AddDataPoint[/]?",    skh.UploadLabeledDatapointsHandler),
                    (r"/ClearDataset[/]?",    skh.ClearDataset),
                    (r"/UpdateModel[/]?",     skh.UpdateModel),
                    (r"/PredictOne[/]?",      skh.PredictOne),
                    (r"/SetParameters[/]?",   skh.SetParameters),
                    (r"/GetParameters[/]?",   skh.GetParameters),
                    ]

        self.handlers_string = str(handlers)

        try:
            self.client  = MongoClient(serverSelectionTimeoutMS=50) # local host, default port
            print(self.client.server_info()) # force pymongo to look for possible running servers, error if none running
            # if we get here, at least one instance of pymongo is running
            self.db = self.client.sklearndatabase # database with labeledinstances, models

        except ServerSelectionTimeoutError as inst:
            print('Could not initialize database connection, stopping execution')
            print('Are you running a valid local-hosted instance of mongodb?')
            #raise inst

        self.clf = {} # the classifier model (in-class assignment, you might need to change this line!)
        # but depending on your implementation, you may not need to change it  ¯\_(ツ)_/¯
        self.KNeighborsNParam = 3
        self.KNeighborsAlgorithmParam = 'auto'
        self.RandomForestNParam = 10

        settings = {'debug':True}
        tornado.web.Application.__init__(self, handlers, **settings)

    def __exit__(self):
        self.client.close() # just in case


def main():
    '''Create server, begin IOLoop
    '''
    tornado.options.parse_command_line()
    http_server = HTTPServer(Application(), xheaders=True)
    http_server.listen(options.port)
    IOLoop.instance().start()

if __name__ == "__main__":
    main()