13 lines
332 B
Python
13 lines
332 B
Python
|
from bayes import Bayes
|
||
|
|
from datapreparator import DataPreparator
|
||
|
|
import pandas as pd
|
||
|
|
import os
|
||
|
|
|
||
|
|
filename = 'music_genre.csv'
|
||
|
|
if os.path.isfile(filename):
|
||
|
|
data = pd.read_csv(filename)
|
||
|
|
else:
|
||
|
|
data_raw = pd.read_csv('music_genre_raw.csv')
|
||
|
|
data = DataPreparator.prepare_data(data_raw)
|
||
|
|
data.to_csv(filename, index=False)
|