add factor history
Browse files- app.py +26 -9
- requirements.txt +1 -1
app.py
CHANGED
@@ -1,13 +1,20 @@
|
|
1 |
import gradio as gr
|
|
|
|
|
|
|
|
|
|
|
|
|
2 |
from fsrs_optimizer import Optimizer
|
3 |
|
4 |
|
5 |
def interface_func(weights: str, ratings: str, request_retention: float) -> str:
|
6 |
-
weights = weights.replace(
|
7 |
optimizer = Optimizer()
|
8 |
-
optimizer.w = list(map(lambda x: float(x.strip()), weights.split(
|
9 |
test_sequence = optimizer.preview_sequence(
|
10 |
-
ratings.replace(
|
|
|
11 |
default_preview = optimizer.preview(request_retention)
|
12 |
return test_sequence, default_preview
|
13 |
|
@@ -15,14 +22,24 @@ def interface_func(weights: str, ratings: str, request_retention: float) -> str:
|
|
15 |
iface = gr.Interface(
|
16 |
fn=interface_func,
|
17 |
inputs=[
|
18 |
-
gr.inputs.Textbox(
|
19 |
-
|
|
|
|
|
|
|
20 |
gr.inputs.Textbox(label="ratings", lines=1, default="3,3,3,3,3,3,3"),
|
21 |
-
gr.inputs.Slider(
|
22 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
23 |
],
|
24 |
-
outputs=[gr.outputs.Textbox(label="test sequences"),
|
25 |
-
gr.outputs.Textbox(label="default preview")]
|
26 |
)
|
27 |
|
28 |
iface.launch()
|
|
|
1 |
import gradio as gr
|
2 |
+
import os
|
3 |
+
import sys
|
4 |
+
|
5 |
+
if os.environ.get("DEV_MODE"):
|
6 |
+
# for local development
|
7 |
+
sys.path.insert(0, os.path.abspath("../fsrs-optimizer/src/fsrs_optimizer/"))
|
8 |
from fsrs_optimizer import Optimizer
|
9 |
|
10 |
|
11 |
def interface_func(weights: str, ratings: str, request_retention: float) -> str:
|
12 |
+
weights = weights.replace("[", "").replace("]", "")
|
13 |
optimizer = Optimizer()
|
14 |
+
optimizer.w = list(map(lambda x: float(x.strip()), weights.split(",")))
|
15 |
test_sequence = optimizer.preview_sequence(
|
16 |
+
ratings.replace(" ", ""), request_retention
|
17 |
+
)
|
18 |
default_preview = optimizer.preview(request_retention)
|
19 |
return test_sequence, default_preview
|
20 |
|
|
|
22 |
iface = gr.Interface(
|
23 |
fn=interface_func,
|
24 |
inputs=[
|
25 |
+
gr.inputs.Textbox(
|
26 |
+
label="weights",
|
27 |
+
lines=1,
|
28 |
+
default="0.4, 0.9, 2.3, 10.9, 4.93, 0.94, 0.86, 0.01, 1.49, 0.14, 0.94, 2.18, 0.05, 0.34, 1.26, 0.29, 2.61",
|
29 |
+
),
|
30 |
gr.inputs.Textbox(label="ratings", lines=1, default="3,3,3,3,3,3,3"),
|
31 |
+
gr.inputs.Slider(
|
32 |
+
label="Your Request Retention",
|
33 |
+
minimum=0.6,
|
34 |
+
maximum=0.97,
|
35 |
+
step=0.01,
|
36 |
+
default=0.9,
|
37 |
+
),
|
38 |
+
],
|
39 |
+
outputs=[
|
40 |
+
gr.outputs.Textbox(label="test sequences"),
|
41 |
+
gr.outputs.Textbox(label="default preview"),
|
42 |
],
|
|
|
|
|
43 |
)
|
44 |
|
45 |
iface.launch()
|
requirements.txt
CHANGED
@@ -1 +1 @@
|
|
1 |
-
FSRS-Optimizer==4.
|
|
|
1 |
+
FSRS-Optimizer==4.15.3
|