| 123456789101112131415161718192021222324 |
- #!flask/bin/python
- from flask import Flask, jsonify, request
- import joblib
- import pandas as pd
- from cassandra_rw import CassandraReadWriteDb
- from PredictTxInfo import PredictTxInfoModel
- app = Flask(__name__)
- cwd = CassandraReadWriteDb(ip_addrs=['172.17.0.2'], keyspace="emp")
- cwd.sync_class_table(PredictTxInfoModel)
- @app.route('/predict/tx', methods=['POST'])
- def create_task():
- tx_data = request.json
- df = pd.DataFrame.from_records([tx_data])
- df = df.drop(['Time'],axis=1)
- model = joblib.load('model3.pipeline')
- tx_data['P'] = model.best_estimator_.predict(df)[0]
- cwd.write_json_table(tx_data)
- tx_data['P'] = str(tx_data['P'])
- return jsonify(tx_data), 201
- if __name__ == '__main__':
- app.run(debug=True)
|
