Python s_prepare_dataloader Exemples

Exemple #1

def test_dataloader_synthetic_channel_len_check(dataframe, station,
                                                target_datetimes,
                                                target_time_offsets,
                                                config_synthetic):
    dl = s_prepare_dataloader(dataframe, target_datetimes, station,
                              target_time_offsets, config_synthetic)

    for data, tgt in dl:
        break
    assert data['images'].shape[-1] == len(config_synthetic["channels"])

Exemple #2

def test_dataloader_synthetic(
    dataframe,
    target_datetimes,
    station,
    target_time_offsets,
    config_synthetic,
):
    dl = s_prepare_dataloader(dataframe, target_datetimes, station,
                              target_time_offsets, config_synthetic)
    assert isinstance(dl, tf.data.Dataset)

Exemple #3

def test_dataloader_synthetic_tgt_ndims(
    dataframe,
    target_datetimes,
    station,
    target_time_offsets,
    config_synthetic,
):
    dl = s_prepare_dataloader(dataframe, target_datetimes, station,
                              target_time_offsets, config_synthetic)
    for img, tgt in dl:
        break
    assert tgt.ndim == 2

Exemple #4

def test_dataloader_synthetic_img_ndims(
    dataframe,
    target_datetimes,
    station,
    target_time_offsets,
    config_synthetic,
):
    dl = s_prepare_dataloader(dataframe, target_datetimes, station,
                              target_time_offsets, config_synthetic)
    for data, tgt in dl:
        break
    assert data['images'].ndim == 5

Exemple #5

def test_dataloader_synthetic_seq_len_check(
    dataframe,
    target_datetimes,
    station,
    target_time_offsets,
    config_synthetic,
):
    dl = s_prepare_dataloader(dataframe, target_datetimes, station,
                              target_time_offsets, config_synthetic)

    for data, tgt in dl:
        break
    assert data['images'].shape[1] == config_synthetic["seq_len"]

Exemple #6

def test_dataloader_synthetic_batch_size_check(
    dataframe,
    target_datetimes,
    station,
    target_time_offsets,
    config_synthetic,
):
    dl = s_prepare_dataloader(dataframe, target_datetimes, station,
                              target_time_offsets, config_synthetic)

    for data, tgt in dl:
        break
    assert data['images'].shape[0] == config_synthetic["batch_size"]
    assert tgt.shape[0] == config_synthetic["batch_size"]