planner.experts.network#

Module Contents#

class planner.experts.network.NetworkSampler(plant: jacta.planner.dynamics.simulator_plant.SimulatorPlant, graph: jacta.planner.core.graph.Graph, params: jacta.planner.core.parameter_container.ParameterContainer, path: str = '', model_name: str = 'actor.pt', state_norm_name: str = 'state_norm.pt')#

Bases: jacta.planner.experts.expert_sampler.ExpertSampler

Parameters:

  • plant (jacta.planner.dynamics.simulator_plant.SimulatorPlant) –

  • graph (jacta.planner.core.graph.Graph) –

  • params (jacta.planner.core.parameter_container.ParameterContainer) –

  • path (str) –

  • model_name (str) –

  • state_norm_name (str) –

callback(node_ids: torch.IntTensor) → torch.FloatTensor#
Parameters:

node_ids (torch.IntTensor) –

Return type:

torch.FloatTensor