def reorder_numpy_dataset()

in spotify_tensorflow/featran.py [0:0]

• 7 lines of code
• 3 McCabe index (conditional complexity)

    def reorder_numpy_dataset(cls,
                              dataset,  # type: Iterator[Dict[str, np.ndarray]]
                              settings_path  # type: str
                              ):
        # type: (...) -> Iterator[OrderedDict[str, np.ndarray]]
        """
        Reorders a numpy dictionary so that feature keys are in the same order as those in a Featran
        settings file.

        :param dataset: A dataset created via Datasets.dict
        :param settings_path: Path to the Featran Settings JSON Directory
        :return: An iterator over an OrderedDict mapping feature names to Numpy arrays
        """
        feature_names = Featran.names(settings_path)
        for batch in dataset:
            yield OrderedDict((name, batch[name]) for name in feature_names)