Class: Informers::Wav2Vec2FeatureExtractor
Instance Attribute Summary
#config
Instance Method Summary
collapse
#initialize
Instance Method Details
#_zero_mean_unit_var_norm(input_values) ⇒ Object
750
751
752
753
754
755
|
# File 'lib/informers/processors.rb', line 750
def _zero_mean_unit_var_norm(input_values)
sum = input_values.sum
mean = sum / input_values.length.to_f
variance = input_values.sum { |b| (b - mean) ** 2 } / input_values.length.to_f
input_values.map { |x| (x - mean) / Math.sqrt(variance + 1e-7) }
end
|
#call(audio) ⇒ Object
757
758
759
760
761
762
763
764
765
766
767
768
769
770
771
772
773
|
# File 'lib/informers/processors.rb', line 757
def call(audio)
input_values = audio
if @config["do_normalize"]
input_values = _zero_mean_unit_var_norm(input_values)
end
{
input_values: [input_values],
attention_mask: [Array.new(input_values.length, 1)]
}
end
|