pqagent.pqmodel module

class pqagent.pqmodel.PqModel(net: pqagent.mlp.MLP)[source]

Bases: object

get_parameters()[source]

get_training_graph(metrics=<class 'list'>, show_plot: bool = True) → figure[source]

classmethod load_model(filepath: str)[source]

property metrics

net: MLP

predict(X: Tensor, *args, **kwargs) → Tensor[source]

result: Result

save(path: str, *args, **kwargs) → None[source]

set_result(ray_result: Result)[source]