File size: 932 Bytes
13c2bf8
4325114
b259c65
13c2bf8
c464a0a
13c2bf8
 
 
 
 
c464a0a
 
13c2bf8
 
b259c65
13c2bf8
 
 
 
 
 
c464a0a
 
 
 
 
 
4325114
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
import gradio
import kaboom
from kaboom.carMakers import runCarDesignProblem

def run_kaboom(reps, steps, comms):
    
    #create a parameters object
    #parameters (p.nAgents = 33, p.nTeams = 11, p.nDims = 56) are automatically set for this problem.
    parameters = kaboom.params.Params()
    parameters.reps = reps
    parameters.pComm = comms
    parameters.steps = steps
    
    #run the simulation with the car designer objective
    team = runCarDesignProblem(parameters)
    
    #check the performance of the team
    #invert score *-1 so that higher score = better performance
    return team.getBestScore()*-1

gradio.Interface(
    fn = run_kaboom, 
    inputs = [
        gradio.Number(label="Number of teams", value=2),
        gradio.Number(label="Steps", value=300),
        gradio.Number(label="Probability of Communication", value=0.2),
    ], outputs = [gradio.Number(label="Performance")]
).launch(debug=True)