import pandas as pd from sklearn.decomposition import NMF N_COMPONENTS = 2 data = pd.read_csv('features/text/tfidf_doc.csv', sep='§') nmf = NMF(n_components=N_COMPONENTS) nmf = pd.DataFrame(nmf.fit_transform(data.drop(['Sequence'], axis='columns'))) nmf = nmf.add_prefix(f'nmf_') nmf = pd.concat([data['Sequence'], nmf], axis='columns') nmf.to_csv(f'features/text/nmf_tfidf_{N_COMPONENTS}.csv', index=False, sep='§')