TypeCare/config.py at main · kupl/TypeCare · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
from pathlib import Path

# Home Path
HOME_PATH = Path.home()
ours = HOME_PATH / "TypeCare"

# Prediction Path
PRED_PATH = ours / "prediction"

# Data Path
DATA_PATH = ours / "data"

# Output Path
OUTPUT_PATH = ours / "output"

# Model Path
MODEL_PATH = PRED_PATH / "typesim_model"


TIGER_TRANSFORM_PATH = PRED_PATH / "Tiger" / "transformed_result.json"
TIGER_OUTPUT_PATH = OUTPUT_PATH / "tiger_result.pkl"


# Configuration for TypeGen
TYPEGEN_PATH= PRED_PATH / "TypeGen"
TYPEGEN_PREDICTION_PATH = TYPEGEN_PATH / "data" / "predictions" / "typegen.json"
TYPEGEN_TRANSFORM_PATH = PRED_PATH / "TypeGen" / "transformed_result.json"
TYPEGEN_OUTPUT_PATH = OUTPUT_PATH / "typegen_result.pkl"

# Configuration for TypeT5
TYPET5_PATH = PRED_PATH / "TypeT5"
TYPET5_PREDICTION_PATH = PRED_PATH / "TypeT5" / 'double-traversal-EvalResultAllTest_0329.pkl'
TYPET5_TRANSFORM_PATH = PRED_PATH / "TypeT5" / "transformed_result.json"
TYPET5_OUTPUT_PATH = OUTPUT_PATH / "typet5_result.pkl"

# Configuration for Example
EXAMPLE_TRANSFORM_PATH = PRED_PATH / "Example" / "transformed_result.json"
EXAMPLE_OUTPUT_PATH = OUTPUT_PATH / "example_result.pkl"