sonusai 0.15.8__py3-none-any.whl → 0.16.0__py3-none-any.whl

{sonusai-0.15.8.dist-info → sonusai-0.16.0.dist-info}/RECORD

@@ -1,27 +1,20 @@
-sonusai/__init__.py,sha256=KmIJ9wni9d9v5pyu0pUxbacZIHGkAywB9CJwl7JME28,1526
+sonusai/__init__.py,sha256=ziSNMMCNS-tIp2ON1e7ST_xNLT-P8BvJfV2IxUvSJZs,2896
 sonusai/aawscd_probwrite.py,sha256=GukR5owp_0A3DrqSl9fHWULYgclNft4D5OkHIwfxxkc,3698
-sonusai/calc_metric_spenh.py,sha256=Cf02uYB6fzDX5anQuBTR_aD7mYt1WgxkEIzZA08uFvs,60825
+sonusai/audiofe.py,sha256=v-GWrma5NNSukkPH427_6ilvFghUlDy4N9sfp_gvwCY,9303
+sonusai/calc_metric_spenh.py,sha256=zJD9cuO4iXvOTuqDjpv-tJQg9bC7r5L7K74Av16lOk4,61762
 sonusai/data/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sonusai/data/genmixdb.yml,sha256=-XSs_hUR6wHJVoTPmSewzXL7u61X-xmHY46lNPatxSE,1025
 sonusai/data/speech_ma01_01.wav,sha256=PK0vMKg-NR6rPE3KouxHGF6PKXnJCr7AwjMqfu98LUA,76644
 sonusai/data/whitenoise.wav,sha256=I2umov0m34y56F9IsIBi1XtE76ZeZaSKDf70cJRe3pI,1920044
-sonusai/data_generator/__init__.py,sha256=ouCpY5EDV35fKFeKGQfIcU8uE-c3QcuNerTxUA1X5L8,232
-sonusai/data_generator/dataset_from_mixdb.py,sha256=Fe_7xxCYURvbZjuIZrvzozdkrx7ZO7gxGg8c5ob2dys,5478
-sonusai/data_generator/keras_from_mixdb.py,sha256=V5CUsGz-akIYdgQy9ABxwNKMYKv01klA4xtMDveF6uI,6167
-sonusai/data_generator/torch_from_mixdb.py,sha256=lvEe9DDu_rIaoyhv9PW4UAnAWp5N74L8kRfxUhsh7oo,4279
 sonusai/doc/__init__.py,sha256=rP5Hgn0Iys_xkuv4caxngdqehuU4zLZsiKuv8Nde67M,19
 sonusai/doc/doc.py,sha256=3z210v6ZckuOlsGZ3ySQBdlCNmBp2M1ahqhqG_eUN58,22664
 sonusai/doc.py,sha256=l8CaFgLI8mqx4tn0aXfxKqa2dy9GgC0zjYxZAkpmi1E,878
-sonusai/evaluate.py,sha256=1BX9dCXMHg6LefbvkevxYpjM3JR-i0jR3Eob2oNh_hQ,9983
-sonusai/genft.py,sha256=CeQN8Sxb_NKeXWJxN9HtzUu687eXl97nHBxzzCzQdLg,5557
+sonusai/genft.py,sha256=6hOds8d-pYRawesLYh7XLrQh4PweWUj8U5Cbzx45bNQ,5572
 sonusai/genmix.py,sha256=0AiUfF7n0CGOp5v-woNfeP3-QaVQUb0CJZc0oXkvqpk,7016
-sonusai/genmixdb.py,sha256=onWkaCPzmUMvDtiGf3A7UdGkOA8xe9zuJTsuLfMdb_s,19597
+sonusai/genmixdb.py,sha256=b-c22X7dd1ePLGrVuhVcn-AXHnxSrC5lTaR6KlrD1-4,19610
 sonusai/gentcst.py,sha256=8jeXirgJe0OGgknC8A-rIudjHeH8UTYPpuh71Ha-I3w,20165
-sonusai/keras_onnx.py,sha256=WHcPIcff4VPdiXqGX-TU-_x-UuKUD3nNpQtCX-2NEVQ,2658
-sonusai/keras_predict.py,sha256=ffrtI12T8_Rm26KsBee0dfEcnEuhOxvtV80YmbJR0m0,9085
-sonusai/keras_train.py,sha256=8_M5vY-CkonPzbOtOF3Vk-wox-42o8fkaOKLjk7Oc2k,13226
-sonusai/lsdb.py,sha256=qTEHQ5X9Ruc0ph4OUhYKFZ6xGC3gyhcNq7t25oyLhc8,5929
-sonusai/main.py,sha256=_s3WCFP_zQ7pD0OgdSVRPQRE38V5Qp6bG56JX-Y79ME,3120
+sonusai/lsdb.py,sha256=TTMQ-0H8fFzUSczt6yjy-9xUjZSdIGQzTVH5Xr6XPSA,5941
+sonusai/main.py,sha256=GC-pQrSqx9tWwIcmEo6V9SraEv5KskBLS_W_wz-f2ZM,2509
 sonusai/metrics/__init__.py,sha256=56itZW3S1I7ZYvbxPmFIVPAh1AIJZdljByz1uCrHqFE,635
 sonusai/metrics/calc_class_weights.py,sha256=dyY7daEIf5Ms5tfTf6wF0fkx_GnMADHOZR_rtsfGoVM,3933
 sonusai/metrics/calc_optimal_thresholds.py,sha256=9fRfwl-aKAbzHJyqGHv4o8BpZXG9HHB7zUJObHXfYM4,3522
@@ -35,24 +28,24 @@ sonusai/metrics/class_summary.py,sha256=4Mb25nuk6eqotnQSFMuOQL3zofGcpNXDfDlPa513
 sonusai/metrics/confusion_matrix_summary.py,sha256=3qg6TMKjJeHtNjj2YnNjPFSlMrQXt0Zcu1dLkGB_aPU,4001
 sonusai/metrics/one_hot.py,sha256=QSeH_GdqBpOAKLrNnQ8gjcPC-vSdUqC0yPEQueTA6VI,13548
 sonusai/metrics/snr_summary.py,sha256=P4U5_Xr7v9F8kF-rZBnpsVNt3p42rIVS6zmch8yfVfg,5575
-sonusai/mixture/__init__.py,sha256=WzPHGSWz6v64HCSFTGboG5o-xBy_0II4i4tkf1UL1Vw,5251
-sonusai/mixture/audio.py,sha256=6W2ihjJGBy7Xggx1imF7bzkymDuipPOgC63j2J7Wf-E,3456
+sonusai/mixture/__init__.py,sha256=BfSJL91URq8-JDlbtpc5SQoQRWEUXvxKozbuquX4Mok,5326
+sonusai/mixture/audio.py,sha256=S-ZROf5rVvwv1TCEuwJHz1FfX4oVubb4QhbybUMMqtM,2150
 sonusai/mixture/augmentation.py,sha256=Blb90tdTwBOj5w9tRcYyS5H67YJuFiXsGqwZWd7ON4g,10468
 sonusai/mixture/class_count.py,sha256=_wFnVl2yEOnbor7pLg7cYOUeX6nioov-03Cv3SEbh2k,996
-sonusai/mixture/config.py,sha256=CXIkVRJmaW2QW_sGl0aIqPf7I_TesyGhUYzxouw5UX4,22266
+sonusai/mixture/config.py,sha256=d2IzZ1samHWGMpoKzSmUwMyAWWhgmyNoxyO8oiUwbsg,22193
 sonusai/mixture/constants.py,sha256=xjCskcQi6khqYZDf7j6z1OkeN1C6wE06kBBapcJiNI4,1428
-sonusai/mixture/datatypes.py,sha256=xN-GdPCEHGE2Ak_TdFbjuSyMs4x7TLRp59trbMTiYLg,8164
+sonusai/mixture/datatypes.py,sha256=zaxfOHw8ddt-i8JPYOPnlqWz_EHBEDoO4q2VAqJViHM,8173
 sonusai/mixture/eq_rule_is_valid.py,sha256=MpQwRA5M76wSiQWEI1lW2cLFdPaMttBLcQp3tWD8efM,1243
-sonusai/mixture/feature.py,sha256=io6OiJAJ3GYvPChiUmPQuP3h0OB2onjYF8o9-AWkmqM,1996
+sonusai/mixture/feature.py,sha256=Rwuf82IoXzhHPGbKYVGcatImF_ssBf_FfvbqghVPXtg,4116
 sonusai/mixture/generation.py,sha256=miUrc3QOSUNIG6mDkiMCZ6M2ulivUZxlYUAJUOVomWc,39039
-sonusai/mixture/helpers.py,sha256=XqpcB15MezEMVJwf3jxzATDJSpj_27b8Cru1TDIFD7w,21326
-sonusai/mixture/log_duration_and_sizes.py,sha256=r-wVjrLW1XBciOL4pkZSYMR7ZNADbojE95TPSQkp3kc,1329
+sonusai/mixture/helpers.py,sha256=GSGSD2KnvOeEIB6IwNTxyaQNjghTSBMB729kUEd_RiM,22403
+sonusai/mixture/log_duration_and_sizes.py,sha256=baTUpqyM15wA125jo9E3posmVJUe3WlpksyO6v9Jul0,1347
 sonusai/mixture/mapped_snr_f.py,sha256=mlbYM1t14OXe_Zg4CjpWTuA_Zun4W0O3bSUXeodRBQs,1845
-sonusai/mixture/mixdb.py,sha256=FQ5hirb2zR8Aj1UNtz89qJQ8wlE0ELC80IxQDmyhsKk,45188
+sonusai/mixture/mixdb.py,sha256=9Pe0mEG8pnEf9NZynTIldc05GfdOrgmcVoIt63RG5DA,45279
 sonusai/mixture/soundfile_audio.py,sha256=Ow_IWIMz4pMsLxMP_JsQ8AuHLCWlYQinLa58CFW97f8,2804
 sonusai/mixture/sox_audio.py,sha256=HT3kYA9TP5QPCuoOJdUMnGVN-qY6q96DGL8zxuog76o,12277
 sonusai/mixture/sox_augmentation.py,sha256=F9tBdNvX2guCn7gRppAFrxRnBtjw9q6qAq2_v_A4hh0,4490
-sonusai/mixture/spectral_mask.py,sha256=LKFnrqZryPHT6FBNiT7yFxOeXc6-AUg6X54N26d8ctg,2107
+sonusai/mixture/spectral_mask.py,sha256=8AkCwhy-PSdP1Uri9miKZP-bXFYnFcH_c9xZCGrHavU,2071
 sonusai/mixture/target_class_balancing.py,sha256=NTNiKZH0_PWLooeow0l41CjJKK8ZTMVbUqz9ZkaNtWk,4900
 sonusai/mixture/targets.py,sha256=wyy5vhLhuN-hqBMBGoziVvEJg3FKFvJFgmEE7_LaV2M,7908
 sonusai/mixture/tokenized_shell_vars.py,sha256=gCxw8SQUcal6mqWKF7hOBTgSQmbJUk1nT0Gn3H8GA0U,4705
@@ -61,68 +54,62 @@ sonusai/mixture/torchaudio_augmentation.py,sha256=1vEDHI0caL1vrgoY2lAWe4CiHE2jKR
 sonusai/mixture/truth.py,sha256=Y41pZ52Xkols9LUler0NlgnilUOscBIucmw4GcxXNzU,1612
 sonusai/mixture/truth_functions/__init__.py,sha256=82lKYHhLy8KW3gHngrocoqwupGVLVsWdIXdYs3vhjOc,359
 sonusai/mixture/truth_functions/crm.py,sha256=_Vy8UMrOUQXsrM3nutvUMWCpvI8GePr01QFlyqLFd4k,2626
-sonusai/mixture/truth_functions/data.py,sha256=NJNZz5fB3jnntUDlnsKJVQIeuHNUvD4x5iNaQVQlo3Y,2857
+sonusai/mixture/truth_functions/data.py,sha256=okFJeOf43NxfdLqWFCBA2pOGqujRlNDYdAcwwR_m8z8,2875
 sonusai/mixture/truth_functions/energy.py,sha256=ydMtMLjMloG76DB30ZHQ5tkBVh4dkMJ82XEhKBokmIk,4281
 sonusai/mixture/truth_functions/file.py,sha256=jOJuC_3y9BH6GGOp9eKcbVrHLVRzUA80BJq59LhcBUM,1539
 sonusai/mixture/truth_functions/phoneme.py,sha256=stYdlPuNytQK_LLT61OJLfYSqKd-sDjQZdtJKGzt5wA,479
 sonusai/mixture/truth_functions/sed.py,sha256=8cHjEFjZaH_0hIOHhPmj4AJz2GpEADM6Ys2x4NoiWSY,2469
-sonusai/mixture/truth_functions/target.py,sha256=3rPXYwU4SBiPP3uIDpOL-B2Xw1Zh3JboD_MYNEyUpuk,5746
-sonusai/mkmanifest.py,sha256=dIPVFKKhnhHdq63OGr6p__pK7fyx3OdKVtbmGUJxsR8,7078
+sonusai/mixture/truth_functions/target.py,sha256=KAsjugDRooOA5BRcHVAbZRgV7l8S5CFg7CZ0XtKZaQ0,5764
+sonusai/mkmanifest.py,sha256=r6bQ6uoBVJIAviqNtJo7AZjXS8EhxdSAW-NHlhPzDCU,8646
 sonusai/mkwav.py,sha256=LZNyhq4gJEs_NtGvRsYHA2qfgkkODpt6HoH1b-Tjjuw,5266
-sonusai/onnx_predict.py,sha256=RhQbbNG3w6rCXuSFUWCaQmUH5JzSP2hmu6TG5_81IVA,9055
-sonusai/plot.py,sha256=GPrbk7SxwojtjR9KLE9jQ6ywYWXca296W0iaRPdnBHI,16982
-sonusai/post_spenh_targetf.py,sha256=Y9J5JbFnCFKCednNiYvVXaC5Z3lF5KxifFJ2RCy4jmg,4975
+sonusai/onnx_predict.py,sha256=Bz_pR28oAZBarNajlKwyzBxmW7ktum77SmxDN2onKPM,9060
+sonusai/plot.py,sha256=u-PvF8guNcm0b-GN99xfEkrcAAtidAEY3RLDzNvcyYk,17014
+sonusai/post_spenh_targetf.py,sha256=NIMhDXeDuUqeWukNaAUMvDw9JpEVCauwjrL2F4M9nrI,4927
 sonusai/queries/__init__.py,sha256=oKY5JeqZ4Cz7DwCwPc1_ydB8bUs6KaMcWFp_w02TjOs,255
 sonusai/queries/queries.py,sha256=FNMUKnoY_Ya9S5sNhsB8ppwy0B7V55ilbbjhQRv_UN8,7552
-sonusai/torchl_onnx.py,sha256=5JYow3XpBaUdtuyAW0mOZyCKL_4FrHvEekYBRdDT6KA,8967
-sonusai/torchl_predict.py,sha256=-wlUdRGPjOvGjCQZY277D8tGeZp1KyDU7TpYH7ovC0c,26657
-sonusai/torchl_train.py,sha256=NPCRB0gwTvabivmOz78gjUreDeO1z16PYuw7L1-pIRQ,9680
 sonusai/tplot.py,sha256=yFyyyg9ymp2Eh-64Muu0EFFEY61MoJSV0a_fy9OWaCk,14485
-sonusai/utils/__init__.py,sha256=ynmdoPJVdbk5eYq9TA9t4QpGEkdiGATDx1pFu43u7YQ,2180
-sonusai/utils/asl_p56.py,sha256=GCKlz-NLInQ0z41XBi0mOvGdSfRZf3WI53necVNDo80,3837
-sonusai/utils/asr.py,sha256=QN1wdO9-EqD72-ixr4lnzsPfT8i0syhTGj1evKNJWe4,2021
-sonusai/utils/asr_functions/__init__.py,sha256=4boXXOXlQHTt8K2DWOwFXSlc8D2NLFd8QTc68yL2ejU,214
-sonusai/utils/asr_functions/aaware_whisper.py,sha256=6JnF8-a-39dYk4gVILWRUD5Ou98T6wPe9g4m8c6MF1A,1994
-sonusai/utils/asr_functions/aixplain_whisper.py,sha256=Fdg3z4HSw1zBu2CQGVU-e8xM3PLTh1Vn4c6fawBOWwQ,1694
-sonusai/utils/asr_functions/data.py,sha256=FQn-y63aOUooOjgKtkzkpFz5Gm9dwn3JPwcR0gC7fYA,418
-sonusai/utils/asr_functions/deepgram.py,sha256=sxFRDJn5EjSLv01G2lMoxXFHzXqFnA9ln4VcQWqGP80,5237
-sonusai/utils/asr_functions/fastwhisper.py,sha256=cyjC3U4UmBZstIvmpuUbNfrcgFFQd1-FLipDaK3_IpU,2856
-sonusai/utils/asr_functions/google.py,sha256=swzEKdod6EwGUKX8LJYcZPf6hJpuvsp9N6QJi7ip0yQ,3320
-sonusai/utils/asr_functions/whisper.py,sha256=IuOodbJY1RsiS3o18cVuBcIXn28TCGQz2BCFR1Up08s,1033
-sonusai/utils/asr_manifest_functions/__init__.py,sha256=Lz12aCGvfngZkLoUxHSqFjHc4wig4ZjQyFbWPWysgrI,310
+sonusai/utils/__init__.py,sha256=TCXlcW8W0Up2f5ciSgz3DabvH1MxrrWD0LK6pQTJkeA,2215
+sonusai/utils/asl_p56.py,sha256=-bvQpd-jRQVURbkZJpRoyEAq6gTv9Rc3oFDbh5_lcjY,3861
+sonusai/utils/asr.py,sha256=6y6VYJizHpuQ3MgKbEQ4t2gofO-MW6Ez23oAd6d23IE,2920
+sonusai/utils/asr_functions/__init__.py,sha256=JyHK67s97bw7QzrlkboWhws4yNytdPatqzLJxfwx-yw,43
+sonusai/utils/asr_functions/aaware_whisper.py,sha256=LzO9CZV0wBWkjmCR2nSWN_AW9UJwriAsC1OYSlfVeT8,1981
+sonusai/utils/asr_manifest_functions/__init__.py,sha256=V-w4R7SHUyoeDuMR3tS12j6DGhmfTlUibMKgq1c6ga0,370
 sonusai/utils/asr_manifest_functions/data.py,sha256=mJsaHccBReguOJu9qsshRhL-3GbeyqM0-PXMseFnZbE,151
 sonusai/utils/asr_manifest_functions/librispeech.py,sha256=HIaytcYmjRUkuR6fCQlv3Jh3IDWSox_A6WFcFFAHN9M,1635
+sonusai/utils/asr_manifest_functions/mcgill_speech.py,sha256=2uF9qgBwcue9W9dlRo16Kvr3jX5g53AGjW2rwob-Cqk,957
 sonusai/utils/asr_manifest_functions/vctk_noisy_speech.py,sha256=-69lM0dz18KbU5_-dmSeqDoNNwgJj4UlxgGkNBEi3wM,2169
+sonusai/utils/audio_devices.py,sha256=LgaXTln1oRArBzaet3rZiIO2plgtaThuGBc3sJ_sLlo,1414
 sonusai/utils/braced_glob.py,sha256=h4hab7YDbM4CjLg9iSzyHZrkd22IPUOY5zZqHdifkh8,1510
-sonusai/utils/calculate_input_shape.py,sha256=fyf8Ggxn9xljJ87BNwIzOh2KD4pKUGFZ2RNzE3195NI,1023
+sonusai/utils/calculate_input_shape.py,sha256=63ILxibYKuTQozY83QN8Y2OOhBEbW_1X47Q0askcHDM,984
 sonusai/utils/convert_string_to_number.py,sha256=i17yIxurp8Iz6NPE-imTRlARrXWqadwm8qbOTuzHZvE,236
+sonusai/utils/create_timestamp.py,sha256=TxoQXWZ3SFdBEHLOv-ujeIsTEJuiFnKOGRy-FQq45YU,148
 sonusai/utils/create_ts_name.py,sha256=8RLKmgXwuGcbDMGgtTuc0MvGFfA7IOVqfjkE2T18GOo,405
 sonusai/utils/dataclass_from_dict.py,sha256=vAGnuMjhy0W9bxZ5usrH7mbQsFog3n0__IC4xyJyVUc,390
 sonusai/utils/db.py,sha256=lI77MJJLs4CTYxhjFUvBom2Kk2imAP34okOeO4irbDc,371
+sonusai/utils/docstring.py,sha256=JBecAq_a7KSzZ04tan0BlFA9SzhOQqY9A7Ue85kFQdU,1446
 sonusai/utils/energy_f.py,sha256=AsoGo3TCMnj9Kzqb9HUAYOL6vPGUseu1p-8z4LG-QYo,1462
 sonusai/utils/engineering_number.py,sha256=6lWRg-XW3FEllh-Zui1WHYcDnsBEY9cqcwcXPIs2-6o,5515
 sonusai/utils/get_frames_per_batch.py,sha256=xnq4tV7MT74N0H6b5ZsiAezqdXucboCLQw1Np9XpZbs,134
 sonusai/utils/get_label_names.py,sha256=bfFV_iFbXdtDp70zAz_CQQN9-QxBw_kt8vyAcr3TITM,828
 sonusai/utils/grouper.py,sha256=qyZ0nj84yOrC-RZsXHC-KJvcUliGktnV8S6-P3PD6_w,203
 sonusai/utils/human_readable_size.py,sha256=SjYT0fUlpbfCzCXHo6csir-VMwqfs5ogr-fgLCEqFk4,279
-sonusai/utils/keras_utils.py,sha256=1FJaaQE2yVYQqB6xTdgzjYKV_yvIA_hKob_uLOLf-0Y,8621
 sonusai/utils/max_text_width.py,sha256=pxiJMwb_zlkNntexgo7S6lAuF7NLLZvFdOCkxdsQJVY,315
+sonusai/utils/model_utils.py,sha256=lt2KOGJqsinG71W0i3U29UXFO-47GMAlEabsf2um7bA,862
 sonusai/utils/numeric_conversion.py,sha256=GRO_2Fba8CcxcFY7bEXKOEUEUX6neA-VN__Bxi1ULsE,340
-sonusai/utils/onnx_utils.py,sha256=A3JyMvjV_0c3YZFEUD8tkT4Er_R4p1P0jjgrfcFXnRQ,3037
+sonusai/utils/onnx_utils.py,sha256=BRsHGlcu5L0v_1z83MNy8TAcBeb7tJd_4OBJgOMLen8,2200
 sonusai/utils/parallel.py,sha256=bxedjCzBv9oxzU7NajRr6mOKmkCWr2P7FWAI0p2p9N8,1981
 sonusai/utils/print_mixture_details.py,sha256=BzYM4-wHHNa6zxPzBMUJxwKt0gKHmvbwdd7Yp0w15Yk,3017
 sonusai/utils/ranges.py,sha256=NPBZOVzMb95GTOIxltVO-wSzgcXqZ14wbdV46JDLKrw,1222
 sonusai/utils/read_mixture_data.py,sha256=Sb30RgSpw6DnH_iD81O7G_KOsdfjQWWLk3euEkxfMa8,453
 sonusai/utils/read_predict_data.py,sha256=5rR_ijrrcS2cKO1Sea2M2QEicokTtW5XtAo6jT5YSX8,1064
-sonusai/utils/reshape.py,sha256=sz-V4Za3DIvqFztTh0yYhQRcofnjc9XBW0CBhxg18lo,5854
+sonusai/utils/reshape.py,sha256=E8Eu6grynaeWwVO6peIR0BF22SrVaJSa1Rkl109lq6Y,5997
 sonusai/utils/seconds_to_hms.py,sha256=oxLuZhTJJr9swj-fOSOrZJ5vBNM7_BrOMQhX1pYpiv0,260
 sonusai/utils/stacked_complex.py,sha256=feLhz3GC1ILxBGMHOj3sJK--sidsXKbfwkalwAVwizc,2950
 sonusai/utils/stratified_shuffle_split.py,sha256=rJNXvBp-GxoKzH3OpL7k0ANSu5xMP2zJ7K1fm_33UzE,7022
-sonusai/utils/trim_docstring.py,sha256=dSrtiRsEN4wkkvKBp6WDr13RUypfqZzgH_jOBLs1ouY,881
-sonusai/utils/wave.py,sha256=TKE-CNPGFXNXUW626CBPzCTNgWJut8I0ZEUsgG9q4Po,586
+sonusai/utils/wave.py,sha256=O4ZXkZ6wjrKGa99wBCdFd8G6bp91MXXDnmGihpaEMh0,856
 sonusai/utils/yes_or_no.py,sha256=eMLXBVH0cEahiXY4W2KNORmwNQ-ba10eRtldh0y4NYg,263
 sonusai/vars.py,sha256=m2AefF0m5bXWGXpJj8Pi42zWL2ydeEj7bkak3GrtMyM,940
-sonusai-0.15.8.dist-info/METADATA,sha256=3eCpCJmXOfr7GV3a7HDWo0iilEVHB5ANdQqS59O0Yi0,2920
-sonusai-0.15.8.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
-sonusai-0.15.8.dist-info/entry_points.txt,sha256=zMNjEphEPO6B3cD1GNpit7z-yA9tUU5-j3W2v-UWstU,92
-sonusai-0.15.8.dist-info/RECORD,,
+sonusai-0.16.0.dist-info/METADATA,sha256=MWrFI-GR33t2hVwlEwtLzzfEZiqO3mekRf7P7L8TaSk,2443
+sonusai-0.16.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+sonusai-0.16.0.dist-info/entry_points.txt,sha256=zMNjEphEPO6B3cD1GNpit7z-yA9tUU5-j3W2v-UWstU,92
+sonusai-0.16.0.dist-info/RECORD,,

{sonusai-0.15.8.dist-info → sonusai-0.16.0.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: poetry-core 1.8.1
+Generator: poetry-core 1.9.0
 Root-Is-Purelib: true
 Tag: py3-none-any

sonusai/data_generator/__init__.py

@@ -1,5 +0,0 @@
-# SonusAI data generator classes
-from .keras_from_mixdb import KerasFromH5
-from .keras_from_mixdb import KerasFromMixtureDatabase
-from .torch_from_mixdb import AawareDataLoader
-from .torch_from_mixdb import TorchFromMixtureDatabase

sonusai/data_generator/dataset_from_mixdb.py

@@ -1,143 +0,0 @@
-import math
-import warnings
-from dataclasses import dataclass
-from typing import Optional
-
-import numpy as np
-
-from sonusai.mixture import GeneralizedIDs
-from sonusai.mixture import MixtureDatabase
-
-with warnings.catch_warnings():
-    warnings.simplefilter('ignore')
-    from keras.utils import Sequence
-
-
-class DatasetFromMixtureDatabase(Sequence):
-    """Generates data for Keras from a SonusAI mixture database
-    """
-
-    @dataclass(frozen=True)
-    class BatchParams:
-        mixids: list[int]
-        offset: int
-        extra: int
-        padding: int
-
-    def __init__(self,
-                 mixdb: MixtureDatabase,
-                 mixids: GeneralizedIDs,
-                 batch_size: int,
-                 timesteps: int,
-                 flatten: bool,
-                 add1ch: bool,
-                 shuffle: bool = False):
-        """Initialization
-        """
-        self.mixdb = mixdb
-        self.mixids = self.mixdb.mixids_to_list(mixids)
-        self.batch_size = batch_size
-        self.timesteps = timesteps
-        self.flatten = flatten
-        self.add1ch = add1ch
-        self.shuffle = shuffle
-        self.stride = self.mixdb.fg_stride
-        self.num_bands = self.mixdb.fg_num_bands
-        self.num_classes = self.mixdb.num_classes
-        self.mixture_frame_segments = None
-        self.batch_frame_segments = None
-        self.total_batches: Optional[int] = None
-
-        self._initialize_mixtures()
-
-    def __len__(self) -> int:
-        """Denotes the number of batches per epoch
-        """
-        return self.total_batches
-
-    def __getitem__(self, batch_index: int) -> tuple[np.ndarray, np.ndarray]:
-        """Get one batch of data
-        """
-        from sonusai.utils import reshape_inputs
-
-        batch_params = self.batch_params[batch_index]
-
-        result = [self.mixdb.mixture_ft(mixid) for mixid in batch_params.mixids]
-        feature = np.vstack([result[i][0] for i in range(len(result))])
-        truth = np.vstack([result[i][1] for i in range(len(result))])
-
-        pad_shape = list(feature.shape)
-        pad_shape[0] = batch_params.padding
-        feature = np.vstack([feature, np.zeros(pad_shape)])
-
-        pad_shape = list(truth.shape)
-        pad_shape[0] = batch_params.padding
-        truth = np.vstack([truth, np.zeros(pad_shape)])
-
-        if batch_params.extra > 0:
-            feature = feature[batch_params.offset:-batch_params.extra]
-            truth = truth[batch_params.offset:-batch_params.extra]
-        else:
-            feature = feature[batch_params.offset:]
-            truth = truth[batch_params.offset:]
-
-        feature, truth = reshape_inputs(feature=feature,
-                                        truth=truth,
-                                        batch_size=self.batch_size,
-                                        timesteps=self.timesteps,
-                                        flatten=self.flatten,
-                                        add1ch=self.add1ch)
-
-        return feature, truth
-
-    def on_epoch_end(self) -> None:
-        """Modification of dataset between epochs
-        """
-        import random
-
-        if self.shuffle:
-            random.shuffle(self.mixids)
-            self._initialize_mixtures()
-
-    def _initialize_mixtures(self) -> None:
-        from sonusai.utils import get_frames_per_batch
-
-        frames_per_batch = get_frames_per_batch(self.batch_size, self.timesteps)
-        # Always extend the number of batches to use all available data
-        # The last batch may need padding
-        self.total_batches = math.ceil(self.mixdb.total_feature_frames(self.mixids) / frames_per_batch)
-
-        # Compute mixid, offset, and extra for dataset
-        # offsets and extras are needed because mixtures are not guaranteed to fall on batch boundaries.
-        # When fetching a new index that starts in the middle of a sequence of mixtures, the
-        # previous feature frame offset must be maintained in order to preserve the correct
-        # data sequence. And the extra must be maintained in order to preserve the correct data length.
-        cumulative_frames = 0
-        start_mixture_index = 0
-        offset = 0
-        self.batch_params = []
-        self.file_indices = []
-        total_frames = 0
-        for idx, mixid in enumerate(self.mixids):
-            current_frames = self.mixdb.mixture(mixid).samples // self.mixdb.feature_step_samples
-            self.file_indices.append(slice(total_frames, total_frames + current_frames))
-            total_frames += current_frames
-            cumulative_frames += current_frames
-            while cumulative_frames >= frames_per_batch:
-                extra = cumulative_frames - frames_per_batch
-                mixids = self.mixids[start_mixture_index:idx + 1]
-                self.batch_params.append(self.BatchParams(mixids=mixids, offset=offset, extra=extra, padding=0))
-                if extra == 0:
-                    start_mixture_index = idx + 1
-                    offset = 0
-                else:
-                    start_mixture_index = idx
-                    offset = current_frames - extra
-                cumulative_frames = extra
-
-        # If needed, add final batch with padding
-        needed_frames = self.total_batches * frames_per_batch
-        padding = needed_frames - total_frames
-        if padding != 0:
-            mixids = self.mixids[start_mixture_index:]
-            self.batch_params.append(self.BatchParams(mixids=mixids, offset=offset, extra=0, padding=padding))

sonusai/data_generator/keras_from_mixdb.py

@@ -1,169 +0,0 @@
-import warnings
-from dataclasses import dataclass
-from multiprocessing import get_context
-from os import cpu_count
-from typing import Optional
-
-import numpy as np
-
-from sonusai.mixture import Feature
-from sonusai.mixture import GeneralizedIDs
-from sonusai.mixture import MixtureDatabase
-from sonusai.mixture import Truth
-
-with warnings.catch_warnings():
-    warnings.simplefilter('ignore')
-    from keras.utils import Sequence
-
-
-@dataclass
-class MPGlobal:
-    mixdb: MixtureDatabase = None
-
-
-MP_GLOBAL = MPGlobal()
-
-
-def _pool_initializer(location: str) -> None:
-    MP_GLOBAL.mixdb = MixtureDatabase(location)
-
-
-def _pool_kernel(mixid: int) -> tuple[Feature, Truth]:
-    return MP_GLOBAL.mixdb.mixture_ft(mixid)
-
-
-class KerasFromMixtureDatabase(Sequence):
-    """Generates data for Keras from a SonusAI mixture database
-    """
-
-    @dataclass(frozen=True)
-    class BatchParams:
-        mixids: list[int]
-        offset: int
-        extra: int
-        padding: int
-
-    def __init__(self,
-                 mixdb: MixtureDatabase,
-                 mixids: GeneralizedIDs,
-                 batch_size: int,
-                 timesteps: int,
-                 flatten: bool,
-                 add1ch: bool,
-                 shuffle: bool = False):
-        """Initialization
-        """
-        self.mixdb = mixdb
-        self.mixids = self.mixdb.mixids_to_list(mixids)
-        self.batch_size = batch_size
-        self.timesteps = timesteps
-        self.flatten = flatten
-        self.add1ch = add1ch
-        self.shuffle = shuffle
-        self.stride = self.mixdb.fg_stride
-        self.num_bands = self.mixdb.fg_num_bands
-        self.num_classes = self.mixdb.num_classes
-        self.mixture_frame_segments: Optional[int] = None
-        self.batch_frame_segments: Optional[int] = None
-        self.total_batches: Optional[int] = None
-
-        self._initialize_mixtures()
-
-        self.pool = get_context('fork').Pool(processes=cpu_count(),
-                                             initializer=_pool_initializer,
-                                             initargs=(self.mixdb.location,))
-
-    def __len__(self) -> int:
-        """Denotes the number of batches per epoch
-        """
-        return self.total_batches
-
-    def __getitem__(self, batch_index: int) -> tuple[np.ndarray, np.ndarray]:
-        """Get one batch of data
-        """
-        from sonusai.utils import reshape_inputs
-
-        batch_params = self.batch_params[batch_index]
-
-        result = self.pool.map(_pool_kernel, batch_params.mixids)
-        feature = np.vstack([result[i][0] for i in range(len(result))])
-        truth = np.vstack([result[i][1] for i in range(len(result))])
-
-        pad_shape = list(feature.shape)
-        pad_shape[0] = batch_params.padding
-        feature = np.vstack([feature, np.zeros(pad_shape)])
-
-        pad_shape = list(truth.shape)
-        pad_shape[0] = batch_params.padding
-        truth = np.vstack([truth, np.zeros(pad_shape)])
-
-        if batch_params.extra > 0:
-            feature = feature[batch_params.offset:-batch_params.extra]
-            truth = truth[batch_params.offset:-batch_params.extra]
-        else:
-            feature = feature[batch_params.offset:]
-            truth = truth[batch_params.offset:]
-
-        feature, truth = reshape_inputs(feature=feature,
-                                        truth=truth,
-                                        batch_size=self.batch_size,
-                                        timesteps=self.timesteps,
-                                        flatten=self.flatten,
-                                        add1ch=self.add1ch)
-
-        return feature, truth
-
-    def on_epoch_end(self) -> None:
-        """Modification of dataset between epochs
-        """
-        import random
-
-        if self.shuffle:
-            random.shuffle(self.mixids)
-            self._initialize_mixtures()
-
-    def _initialize_mixtures(self) -> None:
-        from sonusai.utils import get_frames_per_batch
-
-        frames_per_batch = get_frames_per_batch(self.batch_size, self.timesteps)
-        # Always extend the number of batches to use all available data
-        # The last batch may need padding
-        self.total_batches = int(np.ceil(self.mixdb.total_feature_frames(self.mixids) / frames_per_batch))
-
-        # Compute mixid, offset, and extra for dataset
-        # offsets and extras are needed because mixtures are not guaranteed to fall on batch boundaries.
-        # When fetching a new index that starts in the middle of a sequence of mixtures, the
-        # previous feature frame offset must be maintained in order to preserve the correct
-        # data sequence. And the extra must be maintained in order to preserve the correct data length.
-        cumulative_frames = 0
-        start_mixture_index = 0
-        offset = 0
-        self.batch_params = []
-        self.file_indices = []
-        total_frames = 0
-        for idx, mixid in enumerate(self.mixids):
-            current_frames = self.mixdb.mixture(mixid).samples // self.mixdb.feature_step_samples
-            self.file_indices.append(slice(total_frames, total_frames + current_frames))
-            total_frames += current_frames
-            cumulative_frames += current_frames
-            while cumulative_frames >= frames_per_batch:
-                extra = cumulative_frames - frames_per_batch
-                mixids = self.mixids[start_mixture_index:idx + 1]
-                self.batch_params.append(self.BatchParams(mixids=mixids, offset=offset, extra=extra, padding=0))
-                if extra == 0:
-                    start_mixture_index = idx + 1
-                    offset = 0
-                else:
-                    start_mixture_index = idx
-                    offset = current_frames - extra
-                cumulative_frames = extra
-
-        # If needed, add final batch with padding
-        needed_frames = self.total_batches * frames_per_batch
-        padding = needed_frames - total_frames
-        if padding != 0:
-            mixids = self.mixids[start_mixture_index:]
-            self.batch_params.append(self.BatchParams(mixids=mixids, offset=offset, extra=0, padding=padding))
-
-
-KerasFromH5 = KerasFromMixtureDatabase

sonusai/data_generator/torch_from_mixdb.py

@@ -1,122 +0,0 @@
-from typing import Optional
-
-import numpy as np
-from torch.utils.data import DataLoader
-from torch.utils.data import Dataset
-from torch.utils.data import Sampler
-
-from sonusai.mixture import GeneralizedIDs
-from sonusai.mixture import MixtureDatabase
-
-
-class MixtureDatabaseDataset(Dataset):
-    """Generates a PyTorch dataset from a SonusAI mixture database
-    """
-
-    def __init__(self,
-                 mixdb: MixtureDatabase,
-                 mixids: GeneralizedIDs,
-                 cut_len: int,
-                 flatten: bool,
-                 add1ch: bool,
-                 random_cut: bool = True):
-        """Initialization
-        """
-        self.mixdb = mixdb
-        self.mixids = self.mixdb.mixids_to_list(mixids)
-        self.cut_len = cut_len
-        self.flatten = flatten
-        self.add1ch = add1ch
-        self.random_cut = random_cut
-
-    def __len__(self):
-        return len(self.mixids)
-
-    def __getitem__(self, idx: int) -> tuple[np.ndarray, np.ndarray, int]:
-        """Get data from one mixture
-        """
-        import random
-
-        from sonusai.utils import reshape_inputs
-
-        feature, truth = self.mixdb.mixture_ft(self.mixids[idx])
-        feature, truth = reshape_inputs(feature=feature,
-                                        truth=truth,
-                                        batch_size=1,
-                                        timesteps=0,
-                                        flatten=self.flatten,
-                                        add1ch=self.add1ch)
-
-        length = feature.shape[0]
-
-        if self.cut_len > 0:
-            if length < self.cut_len:
-                feature_final = []
-                truth_final = []
-                for _ in range(self.cut_len // length):
-                    feature_final.append(feature)
-                    truth_final.append(truth)
-                feature_final.append(feature[: self.cut_len % length])
-                truth_final.append(truth[: self.cut_len % length])
-                feature = np.vstack([feature_final[i] for i in range(len(feature_final))])
-                truth = np.vstack([truth_final[i] for i in range(len(truth_final))])
-            else:
-                if self.random_cut:
-                    start = random.randint(0, length - self.cut_len)
-                else:
-                    start = 0
-                feature = feature[start:start + self.cut_len]
-                truth = truth[start:start + self.cut_len]
-
-        return feature, truth, idx
-
-
-class AawareDataLoader(DataLoader):
-    _cut_len: Optional[int] = None
-
-    @property
-    def cut_len(self) -> int:
-        return self._cut_len
-
-    @cut_len.setter
-    def cut_len(self, value: int) -> None:
-        self._cut_len = value
-
-
-def TorchFromMixtureDatabase(mixdb: MixtureDatabase,
-                             mixids: GeneralizedIDs,
-                             batch_size: int,
-                             flatten: bool,
-                             add1ch: bool,
-                             num_workers: int = 0,
-                             cut_len: int = 0,
-                             drop_last: bool = False,
-                             shuffle: bool = False,
-                             random_cut: bool = True,
-                             sampler: Optional[type[Sampler]] = None,
-                             pin_memory: bool = False) -> AawareDataLoader:
-    """Generates a PyTorch dataloader from a SonusAI mixture database
-    """
-    dataset = MixtureDatabaseDataset(mixdb=mixdb,
-                                     mixids=mixids,
-                                     cut_len=cut_len,
-                                     flatten=flatten,
-                                     add1ch=add1ch,
-                                     random_cut=random_cut)
-
-    if sampler is not None:
-        my_sampler = sampler(dataset)
-    else:
-        my_sampler = None
-
-    result = AawareDataLoader(dataset=dataset,
-                              batch_size=batch_size,
-                              pin_memory=pin_memory,
-                              shuffle=shuffle,
-                              sampler=my_sampler,
-                              drop_last=drop_last,
-                              num_workers=num_workers)
-
-    result.cut_len = cut_len
-
-    return result