🚀v2.0

bighuang624 · Apr 7, 2018 · 8f739f7 · 8f739f7
1 parent edf02c6
commit 8f739f7
Show file tree

Hide file tree

Showing 25 changed files with 30,173 additions and 123 deletions.
diff --git a/.gitignore b/.gitignore
@@ -99,3 +99,6 @@ ENV/
 
 # mypy
 .mypy_cache/
+
+.DS_Store
+node_modules/
diff --git a/README.md b/README.md
@@ -3,7 +3,9 @@
 [![Python](https://img.shields.io/badge/python-3.5%2B-green.svg)]()
 [![MIT license](https://img.shields.io/badge/license-MIT-blue.svg)](https://github.com/bighuang624/sentiment-analysis-webapp/blob/master/LICENSE)
 
-情感分析 web 应用 | A web app about sentiment analysis
+中文短文本情感分析 web 应用 | A web app about Chinese sentences sentiment analysis
+
+![example.png](https://raw.githubusercontent.com/bighuang624/sentiment-analysis-webapp/master/docs/example.png)
 
 ## Installation
 
@@ -39,23 +41,26 @@ Open index.html and have fun.  :smiley:
 
 ## Structure
 
-```py
+```
 .
 ├── LICENSE
 ├── README.md
 ├── app.py
 ├── index.html
-├── models  # 存放持久化模型
+├── models
+│   ├── douban_comment.pkl
 │   └── restaurant_comment.pkl
 ├── requirements.txt
 ├── static
-│   ├── main.css
-│   └── main.js
-└── training  # 存放模型训练代码
+└── training
+    ├── douban-comment.csv
+    ├── douban_comment_trainning.py
     ├── restaurant-comment.csv
     └── restaurant_comment_trainning.py
 ```
 
+The source code of the frontend can be found in [sentiment-analysis-webapp-frontend](https://github.com/bighuang624/sentiment-analysis-webapp-frontend), which is powered by Vue.js.
+
 ## Contributors
 
 This work is doing by Siteng Huang, Muzhe Zhou, Ziyi Wu, Zhongchao Cai, Xu Zheng and Suyang Hu.

diff --git a/app.py b/app.py
@@ -2,20 +2,35 @@
 from flask_restful import Resource, Api, abort, reqparse
 from flask_cors import CORS
 from snownlp import SnowNLP
+from sklearn.externals import joblib
+import pandas as pd
+import numpy as np
 
 app = Flask(__name__)
 cors = CORS(app, resources={r"/predict": {"origins": "*"}})
 api = Api(app)
 
+# 读取模型
+restaurant_clf = joblib.load('./models/restaurant_comment.pkl')
+douban_clf = joblib.load('./models/douban_comment.pkl')
+
 # 参数解析
 parser = reqparse.RequestParser()
 parser.add_argument('sentence', type=str)
+parser.add_argument('type', type=str)
 
 class Prediction(Resource):
     def post(self):
         args = parser.parse_args()
-        s = SnowNLP(args['sentence'])
-        return s.sentiments*5, 200
+        if args['type'] == 'restaurant':
+            result = restaurant_clf.predict([args['sentence']]).astype(np.str)
+            return result[0], 200
+        elif args['type'] == 'douban':
+            result = douban_clf.predict([args['sentence']]).astype(np.str)
+            return result[0], 200
+        else:
+            s = SnowNLP(args['sentence'])
+            return s.sentiments*5, 200
 
 api.add_resource(Prediction, '/predict')
 

diff --git a/example.png b/example.png
diff --git a/index.html b/index.html
@@ -1,33 +1 @@
-<!DOCTYPE html>
-<html>
-
-<head>
-    <meta charset="UTF-8">
-    <title>中文短文本情感分析</title>
-    <meta name="description" content="中文短文本情感分析，以 web app 的形式展示。欢迎使用！">
-    <meta name="keywords" content="情感分析,短文本,中文,sentiment-analysis">
-    <meta name="viewport" content="width=device-width, initial-scale=1.0">
-    <meta http-equiv="X-UA-Compatible" content="ie=edge">
-    <link rel="stylesheet" href="./static/main.css">
-</head>
-
-<body>
-    <header>中文短文本情感分析</header>
-    <div class="container">
-        <input type="text" placeholder="请输入短文本">
-        <button id="start">开始预测</button>
-        <button id="random">随便测测</button>
-        <div class="result">
-            <span>情感指数：</span>
-            <span id="score"></span>
-        </div>
-    </div>
-    <footer>
-        <a href="https://github.com/bighuang624/sentiment-analysis-webapp" target="_blank">开源项目地址</a>
-    </footer>
-</body>
-
-</html>
-
-<script src="https://cdn.bootcss.com/jquery/3.3.1/jquery.min.js"></script>
-<script src="./static/main.js"></script>
+<!DOCTYPE html><html><head><meta charset=UTF-8><title>中文短文本情感分析</title><meta name=description content="中文短文本情感分析项目，以 web app 的形式展示。欢迎使用！"><meta name=keywords content=情感分析,短文本,中文,sentiment-analysis><meta name=viewport content="width=device-width,initial-scale=1"><meta http-equiv=X-UA-Compatible content="ie=edge"><link rel="shortcut icon" href=./favicon.ico><link href=./static/css/app.972711d2bd911132aa864d9387b74d24.css rel=stylesheet></head><body><div id=app></div><script type=text/javascript src=./static/js/manifest.3ad1d5771e9b13dbdad2.js></script><script type=text/javascript src=./static/js/vendor.9429d43577f10c9ec8f3.js></script><script type=text/javascript src=./static/js/app.5cf95aae75260ea1d81f.js></script></body></html>
diff --git a/models/.DS_Store b/models/.DS_Store
diff --git a/models/douban_comment.pkl b/models/douban_comment.pkl
diff --git a/static/css/app.972711d2bd911132aa864d9387b74d24.css b/static/css/app.972711d2bd911132aa864d9387b74d24.css
diff --git a/static/css/app.972711d2bd911132aa864d9387b74d24.css.map b/static/css/app.972711d2bd911132aa864d9387b74d24.css.map
diff --git a/static/fonts/ionicons.05acfdb.woff b/static/fonts/ionicons.05acfdb.woff
diff --git a/static/fonts/ionicons.24712f6.ttf b/static/fonts/ionicons.24712f6.ttf
diff --git a/static/fonts/ionicons.2c2ae06.eot b/static/fonts/ionicons.2c2ae06.eot