flask/INFERENCE CALCULATOR.py
2023-06-28 16:05:05 +00:00

55 lines
1.6 KiB
Python

import numpy as np
from flask import Flask
from flask import jsonify
from flask import render_template
from flask import request
app = Flask(__name__)
@app.route("/", methods=["GET", "POST"])
def home():
result = None
if request.method == "POST":
# Get the comma-separated string, split by commas, and convert to integer
s_data = np.array(
[int(i) for i in request.form.get("s_data", "").split(",") if i]
)
ns_data = np.array(
[int(i) for i in request.form.get("ns_data", "").split(",") if i]
)
total = sum(s_data) + sum(ns_data)
# Calculate marginal probabilities
marginal_prob_s = s_data / total
marginal_prob_ns = ns_data / total
# Calculate joint probabilities
joint_prob = np.outer(marginal_prob_s, marginal_prob_ns)
# Calculate likelihoods
posterior_s = s_data / sum(s_data)
posterior_ns = ns_data / sum(ns_data)
# likelihood
likelihood_s = marginal_prob_s
likelihood_ns = marginal_prob_ns
# Return results as dictionary
result = {
"JOINT PROBABILITY": joint_prob.tolist(),
"MARGINAL PROBABILITY_S": marginal_prob_s.tolist(),
"MARGINAL PROBABILITY_NS": marginal_prob_ns.tolist(),
"LIKELIHOOD_S": likelihood_s.tolist(),
"LIKELIHOOD_NS": likelihood_ns.tolist(),
"POSTERIOR_S": posterior_s.tolist(),
"POSTERIOR_NS": posterior_ns.tolist(),
}
return render_template("index.html", result=result)
if __name__ == "__main__":
app.run(debug=True)