JarrettYe commited on
Commit
62714b2
·
verified ·
1 Parent(s): e29554c

add factor history

Browse files
Files changed (2) hide show
  1. app.py +26 -9
  2. 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('[', '').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(' ', ''), request_retention)
 
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(label="weights", lines=1,
19
- default="0.4, 0.6, 2.4, 5.8, 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"),
 
 
 
20
  gr.inputs.Textbox(label="ratings", lines=1, default="3,3,3,3,3,3,3"),
21
- gr.inputs.Slider(label="Your Request Retention",
22
- minimum=0.6, maximum=0.97, step=0.01, default=0.9)
 
 
 
 
 
 
 
 
 
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.12.0
 
1
+ FSRS-Optimizer==4.15.3