The objective of this project is to use given three attriabutes, resonant frequency, electric field and magnetic field to predict height and radius.
- Python >= 3.6 (Recommend to use Anaconda or Miniconda)
- PyTorch >= 1.3
- Pandas -Sklearn
- File: Model1.xlsx
- Objective: Using given "Electric_Abs_3D", "Magnetic_Abs_3D", "Mode 1" to predict "cR" and "cH".
python decision_tree.py
python random_forest.py
python NN_classfication.py
python NN_regression.py
# Linear Regression
linear_reg = linear_model.LinearRegression()
# Decision Tree Regression
decision_tree_reg = tree.DecisionTreeRegressor()
# KNN Regression
knn_reg = neighbors.KNeighborsRegressor()
# Random Forest Regression
random_forest_reg = ensemble.RandomForestRegressor(n_estimators=20)
# Ada boost Regression
ada_boost_reg = ensemble.AdaBoostRegressor(n_estimators=50)
# GBRT Regression
GBRT_reg = ensemble.GradientBoostingRegressor(n_estimators=100)
# Bagging Regression
bagging_reg = BaggingRegressor()
# Extra Regression
extra_tree_reg = ExtraTreeRegressor()