diff --git a/crossai/performance/batch_pilot_evaluation.py b/crossai/performance/batch_pilot_evaluation.py index 8bf6600..17c65b1 100644 --- a/crossai/performance/batch_pilot_evaluation.py +++ b/crossai/performance/batch_pilot_evaluation.py @@ -103,7 +103,7 @@ def batch_evaluate_audio(model, folder, classes: list = None, segments=segments) if pipeline is not None: - eval_object = pipeline.fit_transform(eval_object) + eval_object = pipeline.transform(eval_object) filename = file.replace(file.split(".")[-1], "json") diff --git a/crossai/performance/loader/_audio.py b/crossai/performance/loader/_audio.py index 15c19ea..77bf48e 100644 --- a/crossai/performance/loader/_audio.py +++ b/crossai/performance/loader/_audio.py @@ -35,9 +35,10 @@ def audio_loader(filename, classes, sampling_rate=22500): len(signal), sampling_rate=sampling_rate)[0] # create the object - df = pd.DataFrame(columns=['data', 'label']) + df = pd.DataFrame(columns=['data', 'label', 'filename']) df['data'] = [signal] df['label'] = [labels] + df['filename'] = [filename] crossai_object = Audio(df) diff --git a/crossai/pipelines/timeseries.py b/crossai/pipelines/timeseries.py index 48de3f5..550ddda 100644 --- a/crossai/pipelines/timeseries.py +++ b/crossai/pipelines/timeseries.py @@ -185,10 +185,12 @@ def transform(self, X, y=None): Z.append(unique[np.argmax(counts)]) else: Z.append(np.repeat(X.labels[i], len(Y[i]))) - K.append(np.repeat(X.filename[i], len(Y[i]))) + + K.append(np.repeat(X.filename[i], len(Y[i]))) - X.data = list(np.concatenate(Y, axis=0)) X.filename = list(np.concatenate(K, axis=0)) - X.labels = list(np.concatenate(Z, axis=0)) + X.data = list(np.concatenate(Y, axis=0)) + + X.labels = list(Z) return X