def get_named_ground_truth_data(name):
"""Returns ground truth data set based on name.
Args:
name: String with the name of the dataset.
Raises:
ValueError: if an invalid data set name is provided.
"""
if name == "dsprites_full":
return dsprites.DSprites([1, 2, 3, 4, 5])
elif name == "dsprites_noshape":
return dsprites.DSprites([2, 3, 4, 5])
elif name == "color_dsprites":
return dsprites.ColorDSprites([1, 2, 3, 4, 5])
elif name == "noisy_dsprites":
return dsprites.NoisyDSprites([1, 2, 3, 4, 5])
elif name == "scream_dsprites":
return dsprites.ScreamDSprites([1, 2, 3, 4, 5])
elif name == "smallnorb":
return norb.SmallNORB()
elif name == "cars3d":
return cars3d.Cars3D()
elif name == "shapes3d":
return shapes3d.Shapes3D()
elif name == "dummy_data":
return dummy_data.DummyData()
else:
raise ValueError("Invalid data set name.")
def get_named_ground_truth_data(name,
corr_type='plane',
corr_indices=[3, 4],
col=None):
"""Returns ground truth data set based on name.
Args:
name: String with the name of the dataset.
Raises:
ValueError: if an invalid data set name is provided.
"""
if name == "dsprites_full":
return dsprites.DSprites([1, 2, 3, 4, 5])
elif name == "correlated_dsprites_full":
return dsprites.CorrelatedDSprites([1, 2, 3, 4, 5], corr_indices,
corr_type)
elif name == "dsprites_noshape":
return dsprites.DSprites([2, 3, 4, 5])
elif name == "correlated_dsprites_noshape":
return dsprites.CorrelatedDSprites([2, 3, 4, 5], corr_indices,
corr_type)
elif name == "color_dsprites":
return dsprites.ColorDSprites([1, 2, 3, 4, 5])
elif name == "backgroundcolor_dsprites":
return dsprites.BackgroundColorDSprites([1, 2, 3, 4, 5, 6], col)
elif name == "correlated_backgroundcolor_dsprites":
return dsprites.CorrelatedBackgroundColorDSprites([1, 2, 3, 4, 5, 6],
corr_indices,
corr_type)
elif name == "correlated_color_dsprites":
return dsprites.CorrelatedColorDSprites([1, 2, 3, 4, 5], corr_indices,
corr_type)
elif name == "noisy_dsprites":
return dsprites.NoisyDSprites([1, 2, 3, 4, 5])
elif name == "correlated_noisy_dsprites":
return dsprites.CorrelatedNoisyDSprites([1, 2, 3, 4, 5], corr_indices,
corr_type)
elif name == "scream_dsprites":
return dsprites.ScreamDSprites([1, 2, 3, 4, 5])
elif name == "correlated_scream_dsprites":
return dsprites.CorrelatedScreamDSprites([1, 2, 3, 4, 5], corr_indices,
corr_type)
elif name == "smallnorb":
return norb.SmallNORB()
elif name == "cars3d":
return cars3d.Cars3D()
elif name == "shapes3d":
return shapes3d.Shapes3D()
elif name == "dummy_data":
return dummy_data.DummyData()
else:
raise ValueError("Invalid data set name: " + name + ".")
def get_dlib_data(task):
ut.log("Loading {}".format(task))
if task == "dsprites":
# 5 factors
return dsprites.DSprites(list(range(1, 6)))
elif task == "shapes3d":
# 6 factors
return shapes3d.Shapes3D()
elif task == "norb":
# 4 factors + 1 nuisance (which we'll handle via n_dim=2)
return norb.SmallNORB()
elif task == "cars3d":
# 3 factors
return cars3d.Cars3D()
elif task == "scream":
# 5 factors + 2 nuisance (handled as n_dim=2)
return dsprites.ScreamDSprites(list(range(1, 6)))
def get_named_ground_truth_data(name):
"""Returns ground truth data set based on name.
Args:
name: String with the name of the dataset.
Raises:
ValueError: if an invalid data set name is provided.
"""
if name == "dsprites_full":
return dsprites.DSprites([1, 2, 3, 4, 5])
elif name == "dsprites_noshape":
return dsprites.DSprites([2, 3, 4, 5])
elif name == "color_dsprites":
return dsprites.ColorDSprites([1, 2, 3, 4, 5])
elif name == "noisy_dsprites":
return dsprites.NoisyDSprites([1, 2, 3, 4, 5])
elif name == "scream_dsprites":
return dsprites.ScreamDSprites([1, 2, 3, 4, 5])
elif name == "smallnorb":
return norb.SmallNORB()
elif name == "cars3d":
return cars3d.Cars3D()
elif name == "mpi3d_toy":
return mpi3d.MPI3D(mode="mpi3d_toy")
elif name == "mpi3d_realistic":
return mpi3d.MPI3D(mode="mpi3d_realistic")
elif name == "mpi3d_real":
return mpi3d.MPI3D(mode="mpi3d_real")
elif name == "shapes3d":
return shapes3d.Shapes3D()
elif name == "dummy_data":
return dummy_data.DummyData()
elif name == "modelnet":
return DisLibGroundTruthData(**MODELNET_PARAMS)
elif name == "arrow":
return DisLibGroundTruthData(**ARROW_PARAMS)
elif name == "pixel4":
return DisLibGroundTruthData(**WRAPPED_PIXEL4_PARAMS)
elif name == "pixel8":
return DisLibGroundTruthData(**WRAPPED_PIXEL8_PARAMS)
else:
raise ValueError("Invalid data set name.")
def get_named_ground_truth_data(name):
"""Returns ground truth data set based on name.
Args:
name: String with the name of the dataset.
Raises:
ValueError: if an invalid data set name is provided.
"""
if name == "threeDotsCache":
# a large random sample from ThreeDots
return threeDots.ThreeDotsTrainingCache()
elif name == "threeDots":
return threeDots.ThreeDots()
elif name == "dsprites_full":
return dsprites.DSprites([1, 2, 3, 4, 5])
elif name == "dsprites_noshape":
return dsprites.DSprites([2, 3, 4, 5])
elif name == "color_dsprites":
return dsprites.ColorDSprites([1, 2, 3, 4, 5])
elif name == "noisy_dsprites":
return dsprites.NoisyDSprites([1, 2, 3, 4, 5])
elif name == "scream_dsprites":
return dsprites.ScreamDSprites([1, 2, 3, 4, 5])
elif name == "smallnorb":
return norb.SmallNORB()
elif name == "cars3d":
return cars3d.Cars3D()
elif name == "mpi3d_toy":
return mpi3d.MPI3D(mode="mpi3d_toy")
elif name == "mpi3d_realistic":
return mpi3d.MPI3D(mode="mpi3d_realistic")
elif name == "mpi3d_real":
return mpi3d.MPI3D(mode="mpi3d_real")
elif name == "mpi3d_multi_real":
return mpi3d_multi.MPI3DMulti(mode="mpi3d_real")
elif name == "shapes3d":
return shapes3d.Shapes3D()
elif name == "dummy_data":
return dummy_data.DummyData()
else:
raise ValueError("Invalid data set name.")
def get_named_ground_truth_data(name):
"""Returns ground truth data set based on name.
Args:
name: String with the name of the dataset.
Raises:
ValueError: if an invalid data set name is provided.
"""
if name == "dsprites_full":
return dsprites.DSprites([1, 2, 3, 4, 5])
elif name == "dsprites_noshape":
return dsprites.DSprites([2, 3, 4, 5])
elif name == "color_dsprites":
return dsprites.ColorDSprites([1, 2, 3, 4, 5])
elif name == "noisy_dsprites":
return dsprites.NoisyDSprites([1, 2, 3, 4, 5])
elif name == "scream_dsprites":
return dsprites.ScreamDSprites([1, 2, 3, 4, 5])
elif name == "smallnorb":
return norb.SmallNORB()
elif name == "cars3d":
return cars3d.Cars3D()
elif name == "mpi3d_toy":
return mpi3d.MPI3D(mode="mpi3d_toy")
elif name == "mpi3d_realistic":
return mpi3d.MPI3D(mode="mpi3d_realistic")
elif name == "mpi3d_real":
return mpi3d.MPI3D(mode="mpi3d_real")
elif name == "3dshapes":
return shapes3d.Shapes3D()
elif name == "3dshapes_holdout" or name == "3dshapes_pca_holdout_s5000":
return shapes3d_partial.Shapes3DPartial(name)
elif name == "3dshapes_model_all":
return shapes3d_partial.Shapes3DPartial(name), None
elif name[:8] == "3dshapes":
return shapes3d_partial.Shapes3DPartial(
name + '_train'), shapes3d_partial.Shapes3DPartial(name + '_valid')
elif name == "dummy_data":
return dummy_data.DummyData()
else:
raise ValueError("Invalid data set name.")