Как обработать загруженный CSV-файл с помощью web.py в Python - PullRequest
Новая
       20

Как обработать загруженный CSV-файл с помощью web.py в Python

0 голосов
/ 04 января 2019

Я пытаюсь создать API для моей модели.

Я пытаюсь загрузить файл CSV, затем прочитать данные в CSV, а затем использовать модель для прогнозирования в API.

Я могу загрузить файл и сохранить в пути, но не могу прочитать данные CSV для прогнозирования, используя web.py в python

Я сохранил модель для прогнозирования и затем загрузил в этот кодпрогнозные данные.

upload.py 

import web
from sklearn.externals import joblib
import requests

urls = ('/upload', 'Upload')

class Upload:

    def GET(self):

        web.header("Content-Type","text/html; charset=utf-8")

        return """<html><head></head><body>
                <form method="POST" enctype="multipart/form-data" action="">
                <input type="file" name="myfile" />
                <br/>
                <br/>
                <input type="submit" />
                </form>
                </body></html>"""    

    def POST(self):

        x = web.input(myfile=[])
        filedir = 'D:/API_CITY_PRED/Upload' # change this to the directory you want to store the file in.
        svmModel = open('D:/Model/model_city_id_predictor.pkl', 'rb')
        svmModel = joblib.load(svmModel)
        class_prediced = svmModel.predict(x)
        output = "Predicted City ID: " + str(class_prediced)
        print (output)

        if 'myfile' in x: # to check if the file-object is created
            filepath=x.myfile.filename.replace('\\','/') # replaces the windows-style slashes with linux ones.
            filename=filepath.split('/')[-1] # splits the and chooses the last part (the filename with extension)
            fout = open(filedir +'/'+ filename,'wb') # creates the file where the uploaded file should be stored
            fout.write(x.myfile.file.read()) # writes the uploaded file to the newly created file.
            fout.close() # closes the file, upload complete.

        return output
        raise web.seeother('/upload')

if __name__ == "__main__":
   app = web.application(urls, globals()) 
   app.run()

Edit-1

# x is the input data

svmModel = open('D:/Model/model_city_id_predictor.pkl', 'rb') # SVM Model Imported svmModel = joblib.load(svmModel) # Model Loaded class_prediced = svmModel.predict(x) # here we are using to predict

Пожалуйста, предложите

1 Ответ

0 голосов
/ 11 февраля 2019

Использование 

x = web.input(file={})
fout = open('path/to/location/', 'wb')  # creates the file where the uploaded file should be stored
fout.write(x.file.file.read())  # writes the uploaded file to the newly created file.
fout.close()

Это запишет ваш файл в указанное место.

...