feature extractor finished but not checked
working on inference