23 lines
777 B
Python
23 lines
777 B
Python
import argparse
|
|
from core import recommendation
|
|
from utils import pg_client
|
|
import asyncio
|
|
|
|
#TODO: remove this module
|
|
import pandas as pd
|
|
|
|
parser = argparse.ArgumentParser(description='Handle machine learning inputs.')
|
|
parser.add_argument('--mode', choices=['train', 'test'], required=True, help='--mode sets the model in train or test mode')
|
|
parser.add_argument('--kernel', default='linear', help='--kernel set the kernel to be used for SVM')
|
|
|
|
args = parser.parse_args()
|
|
|
|
if __name__ == '__main__':
|
|
asyncio.run(pg_client.init())
|
|
data1 = recommendation.query_funnels()
|
|
print(pd.DataFrame(data1))
|
|
data2 = recommendation.query_with_filters()
|
|
print(pd.DataFrame(data2))
|
|
data3 = recommendation.query_metrics()
|
|
print(pd.DataFrame(data3))
|
|
print(args)
|