beamds.beam.writer package

Submodules

beamds.beam.writer.mlflow module

beamds.beam.writer.writer module

class beamds.beam.writer.writer.BeamWriter(*args, _store_init_path=None, _save_init_args=True, **kwargs)

Bases: BeamBase

add_audio(tag, snd_tensor, global_step=None, sample_rate=44100, walltime=None)

add_custom_scalars(layout)

add_custom_scalars_marginchart(tags: List[str], category: str = 'default', title: str = 'untitled')

add_custom_scalars_multilinechart(tags: List[str], category: str = 'default', title: str = 'untitled')

add_embedding(mat, metadata=None, label_img=None, global_step=None, tag='default', metadata_header=None)

add_figure(tag, figure, global_step=None, close=True, walltime=None)

add_graph(model, input_to_model=None, verbose=False)

add_histogram(tag, values, global_step=None, bins='tensorflow', walltime=None, max_bins=None)

add_histogram_raw(tag, min, max, num, sum, sum_squares, bucket_limits, bucket_counts, global_step=None, walltime=None)

add_hparams(hparam_dict, metric_dict, **kwargs)

add_image(tag, img_tensor, global_step=None, walltime=None, dataformats='CHW')

add_images(tag, img_tensor, global_step=None, walltime=None, dataformats='NCHW')

add_mesh(tag, vertices, colors=None, faces=None, config_dict=None, global_step=None)

add_onnx_graph(model)

add_openvino_graph(model)

add_pr_curve(tag, labels, predictions, global_step=None, num_thresholds=127, weights=None, walltime=None)

add_pr_curve_raw(tag, true_positive_counts, false_positive_counts, true_negative_counts, false_negative_counts, precision, recall, global_step=None, num_thresholds=127, weights=None, walltime=None)

add_scalar(tag, scalar_value, global_step=None)

add_scalars(main_tag, tag_scalar_dict, global_step=None)

add_tag(tag)

add_text(tag, text_string, global_step=None, walltime=None)

add_video(tag, vid_tensor, global_step=None, fps=4, walltime=None)

property client

classmethod from_url(url)

property root

set_name(name)

class beamds.beam.writer.writer.CometWriter(*args, _store_init_path=None, _save_init_args=True, **kwargs)

Bases: BeamWriter

add_graph(model, input_to_model=None, verbose=False)

add_tag(tag)

property client

classmethod from_url(url)

property root

set_name(name)

class beamds.beam.writer.writer.TensorboardWriter(*args, _store_init_path=None, _save_init_args=True, **kwargs)

Bases: BeamWriter

add_tag(tag)

property client

close()

classmethod from_url(url)

property root

set_name(name)

Module contents