first try
This commit is contained in:
parent
242c10da9f
commit
bb5d311df5
462
dev-0/out.tsv
Normal file
462
dev-0/out.tsv
Normal file
@ -0,0 +1,462 @@
|
||||
753233.5625
|
||||
717278.0
|
||||
761310.75
|
||||
737726.625
|
||||
581629.3125
|
||||
633398.5625
|
||||
734318.5
|
||||
742838.8125
|
||||
635378.125
|
||||
736022.625
|
||||
626772.6875
|
||||
726735.5
|
||||
755602.1875
|
||||
660751.5625
|
||||
752824.625
|
||||
737743.6875
|
||||
728132.8125
|
||||
1178018.125
|
||||
735153.5
|
||||
534563.3125
|
||||
741015.5
|
||||
903060.0
|
||||
739771.5
|
||||
954096.4375
|
||||
753063.125
|
||||
688970.6875
|
||||
977016.0
|
||||
737845.9375
|
||||
751359.125
|
||||
768007.75
|
||||
727502.3125
|
||||
730910.4375
|
||||
761072.25
|
||||
647391.75
|
||||
761072.25
|
||||
909365.0
|
||||
789121.0
|
||||
577675.875
|
||||
721061.0
|
||||
793636.75
|
||||
727894.25
|
||||
942082.875
|
||||
891472.375
|
||||
739243.25
|
||||
784162.1875
|
||||
577062.4375
|
||||
624591.5
|
||||
644341.5
|
||||
725832.375
|
||||
624216.5625
|
||||
620229.0625
|
||||
899992.625
|
||||
716937.125
|
||||
747951.0
|
||||
776238.3125
|
||||
729206.375
|
||||
636997.0
|
||||
704648.0
|
||||
687266.625
|
||||
526043.0
|
||||
707684.125
|
||||
582992.5625
|
||||
613992.25
|
||||
1065286.125
|
||||
802480.75
|
||||
657002.625
|
||||
845082.1875
|
||||
780805.1875
|
||||
752364.5
|
||||
682154.5
|
||||
705690.375
|
||||
715948.8125
|
||||
739430.75
|
||||
919589.3125
|
||||
725951.625
|
||||
985195.5
|
||||
764139.5
|
||||
970915.5
|
||||
730160.625
|
||||
943786.875
|
||||
895732.5
|
||||
538721.1875
|
||||
724094.1875
|
||||
725798.25
|
||||
899992.625
|
||||
738646.875
|
||||
532859.25
|
||||
754767.25
|
||||
712165.8125
|
||||
534563.3125
|
||||
794128.0
|
||||
777976.5
|
||||
594205.25
|
||||
942082.875
|
||||
886360.25
|
||||
911921.0625
|
||||
880225.625
|
||||
754750.1875
|
||||
737897.0625
|
||||
547684.5
|
||||
951114.375
|
||||
732614.5
|
||||
738749.125
|
||||
724043.125
|
||||
526434.9375
|
||||
525617.0
|
||||
952818.375
|
||||
724366.875
|
||||
724486.125
|
||||
733466.5
|
||||
732273.6875
|
||||
774173.5
|
||||
895851.8125
|
||||
752040.75
|
||||
724298.6875
|
||||
732273.6875
|
||||
735681.75
|
||||
738067.5
|
||||
720345.25
|
||||
570348.5
|
||||
709609.75
|
||||
939185.9375
|
||||
1076822.5
|
||||
824463.125
|
||||
744542.875
|
||||
701444.375
|
||||
717448.375
|
||||
663409.875
|
||||
728712.1875
|
||||
758175.3125
|
||||
642961.1875
|
||||
1082383.625
|
||||
945965.1875
|
||||
740282.75
|
||||
734829.75
|
||||
886258.0
|
||||
529451.125
|
||||
894999.75
|
||||
717278.0
|
||||
741475.5625
|
||||
636465.875
|
||||
787144.25
|
||||
580402.375
|
||||
609218.0
|
||||
764991.5625
|
||||
536437.75
|
||||
658297.6875
|
||||
682648.6875
|
||||
648260.8125
|
||||
971563.0
|
||||
897538.8125
|
||||
737266.5625
|
||||
992863.75
|
||||
736022.625
|
||||
967575.5
|
||||
717278.0
|
||||
907916.5
|
||||
763117.125
|
||||
727195.625
|
||||
663409.875
|
||||
735187.625
|
||||
675338.25
|
||||
735937.375
|
||||
730910.4375
|
||||
737845.9375
|
||||
708042.0
|
||||
978924.5625
|
||||
888916.3125
|
||||
763287.5
|
||||
613992.25
|
||||
724043.125
|
||||
552166.1875
|
||||
737743.6875
|
||||
753063.125
|
||||
705789.75
|
||||
554228.125
|
||||
951404.0
|
||||
710666.25
|
||||
739345.5
|
||||
730041.375
|
||||
698104.4375
|
||||
548195.75
|
||||
514114.625
|
||||
713733.5
|
||||
700953.125
|
||||
901696.75
|
||||
738646.875
|
||||
613259.5
|
||||
762946.6875
|
||||
742838.8125
|
||||
803523.125
|
||||
770069.625
|
||||
727502.3125
|
||||
595738.875
|
||||
728320.25
|
||||
771807.75
|
||||
732614.5
|
||||
745531.25
|
||||
759368.125
|
||||
776919.9375
|
||||
727502.3125
|
||||
880140.375
|
||||
708076.0625
|
||||
753063.125
|
||||
904116.5
|
||||
724094.1875
|
||||
742838.8125
|
||||
632736.875
|
||||
662285.1875
|
||||
723634.125
|
||||
754767.25
|
||||
778624.0
|
||||
677601.75
|
||||
534563.3125
|
||||
763287.5
|
||||
555898.0625
|
||||
743520.4375
|
||||
783054.5
|
||||
739788.5625
|
||||
732614.5
|
||||
740197.5
|
||||
546679.125
|
||||
503890.28125
|
||||
556443.375
|
||||
548076.4375
|
||||
734318.5
|
||||
589076.0625
|
||||
717278.0
|
||||
730041.375
|
||||
707735.25
|
||||
721435.875
|
||||
708416.875
|
||||
724094.1875
|
||||
629328.75
|
||||
762810.375
|
||||
583742.375
|
||||
734318.5
|
||||
1117450.125
|
||||
723412.5625
|
||||
634440.9375
|
||||
538244.0625
|
||||
526179.3125
|
||||
633827.4375
|
||||
705349.625
|
||||
554500.75
|
||||
733125.6875
|
||||
733636.875
|
||||
639893.875
|
||||
737556.25
|
||||
665284.3125
|
||||
743861.25
|
||||
723276.25
|
||||
708076.0625
|
||||
802855.6875
|
||||
729632.375
|
||||
733602.875
|
||||
739430.75
|
||||
642719.75
|
||||
697491.0
|
||||
663409.875
|
||||
619513.375
|
||||
528939.875
|
||||
594546.0625
|
||||
703798.875
|
||||
552967.125
|
||||
805888.875
|
||||
523060.90625
|
||||
612816.4375
|
||||
1039725.25
|
||||
710938.875
|
||||
707053.625
|
||||
919418.875
|
||||
1036317.125
|
||||
696249.9375
|
||||
747030.8125
|
||||
663665.5
|
||||
677059.375
|
||||
670226.125
|
||||
736022.625
|
||||
992522.9375
|
||||
912773.125
|
||||
947467.6875
|
||||
522634.90625
|
||||
747899.875
|
||||
592501.1875
|
||||
717278.0
|
||||
640626.625
|
||||
521459.09375
|
||||
761583.4375
|
||||
718641.25
|
||||
1005473.75
|
||||
749229.0
|
||||
734318.5
|
||||
920373.1875
|
||||
908683.375
|
||||
700595.25
|
||||
728081.6875
|
||||
740282.75
|
||||
735698.8125
|
||||
636145.0
|
||||
923849.4375
|
||||
792648.375
|
||||
961338.6875
|
||||
804627.875
|
||||
717278.0
|
||||
645176.5
|
||||
553188.625
|
||||
520930.84375
|
||||
573092.0
|
||||
544872.8125
|
||||
733926.625
|
||||
522345.21875
|
||||
706181.6875
|
||||
720305.375
|
||||
570348.5
|
||||
717278.0
|
||||
743690.875
|
||||
772370.125
|
||||
616088.25
|
||||
768399.625
|
||||
904440.25
|
||||
976675.1875
|
||||
580572.8125
|
||||
532859.25
|
||||
544787.625
|
||||
713921.0
|
||||
1184289.0
|
||||
726411.75
|
||||
910217.0
|
||||
886019.375
|
||||
732614.5
|
||||
704548.6875
|
||||
956888.1875
|
||||
760901.8125
|
||||
599147.0
|
||||
1111295.625
|
||||
753063.125
|
||||
917033.25
|
||||
735085.375
|
||||
892051.75
|
||||
773136.9375
|
||||
949188.75
|
||||
557482.875
|
||||
729206.375
|
||||
606324.0
|
||||
783906.5625
|
||||
606324.0
|
||||
570246.25
|
||||
745585.25
|
||||
732819.0
|
||||
728422.5
|
||||
738578.6875
|
||||
533336.375
|
||||
742458.125
|
||||
611197.5625
|
||||
538244.0625
|
||||
717295.0
|
||||
616497.1875
|
||||
712063.5625
|
||||
651481.5
|
||||
760561.0
|
||||
599147.0
|
||||
669183.75
|
||||
906808.875
|
||||
753233.5625
|
||||
560124.125
|
||||
746076.5
|
||||
1146211.75
|
||||
570246.25
|
||||
734301.5
|
||||
739430.75
|
||||
708842.875
|
||||
724946.25
|
||||
845593.375
|
||||
580061.5625
|
||||
722816.125
|
||||
627624.6875
|
||||
906808.875
|
||||
726343.5625
|
||||
660001.75
|
||||
541652.1875
|
||||
915138.8125
|
||||
615642.25
|
||||
733994.75
|
||||
636806.625
|
||||
895831.875
|
||||
558249.6875
|
||||
703884.125
|
||||
740569.5
|
||||
704071.5625
|
||||
540646.75
|
||||
539573.1875
|
||||
771807.75
|
||||
574438.1875
|
||||
556716.0
|
||||
636806.625
|
||||
675338.25
|
||||
574659.75
|
||||
895050.875
|
||||
749859.5
|
||||
743486.375
|
||||
733276.1875
|
||||
1011949.125
|
||||
969518.125
|
||||
936459.5
|
||||
1255859.375
|
||||
714173.6875
|
||||
566940.375
|
||||
725713.0625
|
||||
654889.5625
|
||||
807490.75
|
||||
613157.25
|
||||
580794.3125
|
||||
634492.0
|
||||
797453.8125
|
||||
598499.5
|
||||
992673.375
|
||||
744031.625
|
||||
580760.25
|
||||
591700.3125
|
||||
895542.1875
|
||||
724043.125
|
||||
595298.75
|
||||
764395.125
|
||||
731114.875
|
||||
546662.0625
|
||||
738698.0
|
||||
738698.0
|
||||
733279.0625
|
||||
738425.3125
|
||||
796669.9375
|
||||
795920.1875
|
||||
644324.4375
|
||||
689822.75
|
||||
663069.0625
|
||||
787076.125
|
||||
992352.5
|
||||
617380.4375
|
||||
835519.5
|
||||
736056.6875
|
||||
559749.25
|
||||
630893.5625
|
||||
580095.6875
|
||||
638019.4375
|
||||
654395.4375
|
||||
675048.5625
|
||||
868899.125
|
||||
763696.5
|
||||
714074.375
|
||||
730893.375
|
||||
713852.8125
|
||||
585667.9375
|
||||
713852.8125
|
||||
733347.25
|
||||
574012.1875
|
||||
776545.0625
|
||||
597408.875
|
||||
776545.0625
|
||||
557107.9375
|
||||
744815.5
|
||||
1171488.75
|
||||
775045.5
|
|
37
predict.py
Normal file
37
predict.py
Normal file
@ -0,0 +1,37 @@
|
||||
import pickle
|
||||
import sys
|
||||
import torch
|
||||
import pandas as pd
|
||||
|
||||
def read_data_file(filepath):
|
||||
df = pd.read_csv(filepath, sep='\t', header=None, index_col=None)
|
||||
dataframe = df.iloc[:, [7,10]]
|
||||
dataframe.columns = ['biggy','type']
|
||||
#print(dataframe.size[0])
|
||||
# for x in range(len(dataframe)):
|
||||
# dataframe['biggy'].loc[x] = dataframe['biggy'].loc[x].replace(" ","")
|
||||
#such dumb solution, well, but at least it works
|
||||
dataframe['bias'] = 1
|
||||
dataframe['biggy'] = dataframe['biggy'].astype(float)
|
||||
return dataframe
|
||||
|
||||
def dataframe_to_arrays(dataframe):
|
||||
dataframe1 = dataframe.copy(deep=True)
|
||||
dataframe1["type"] = dataframe1["type"].astype('category').cat.codes
|
||||
return dataframe1
|
||||
|
||||
PREDICT_FILE_PATH = 'test-A/in.tsv'
|
||||
|
||||
def main():
|
||||
w = pickle.load(open('model.pkl', 'rb'))
|
||||
|
||||
data = read_data_file(PREDICT_FILE_PATH)
|
||||
data = dataframe_to_arrays(data)
|
||||
|
||||
for index, row in data.iterrows():
|
||||
#print(row[0], row[1])
|
||||
x = torch.tensor([float(row[0]), float(row[1]), 1])
|
||||
y = x @ w
|
||||
print(y.item())
|
||||
|
||||
main()
|
418
test-A/out.tsv
Normal file
418
test-A/out.tsv
Normal file
@ -0,0 +1,418 @@
|
||||
758158.25
|
||||
761583.4375
|
||||
739686.3125
|
||||
639161.125
|
||||
728115.75
|
||||
810830.625
|
||||
545639.625
|
||||
880205.6875
|
||||
1147683.0
|
||||
735784.0
|
||||
756454.25
|
||||
758175.3125
|
||||
735000.125
|
||||
717278.0
|
||||
734318.5
|
||||
736022.625
|
||||
806911.3125
|
||||
834196.125
|
||||
703645.5
|
||||
939049.625
|
||||
778624.0
|
||||
705349.625
|
||||
646369.3125
|
||||
718982.0
|
||||
618062.0625
|
||||
735698.8125
|
||||
760987.0
|
||||
1111420.75
|
||||
607704.25
|
||||
735681.75
|
||||
1005644.125
|
||||
944298.125
|
||||
1056765.75
|
||||
620621.0
|
||||
636145.0
|
||||
776238.3125
|
||||
922145.375
|
||||
747951.0
|
||||
725798.25
|
||||
681779.625
|
||||
761072.25
|
||||
563344.8125
|
||||
532007.1875
|
||||
754767.25
|
||||
715573.9375
|
||||
738646.875
|
||||
756641.6875
|
||||
713409.75
|
||||
704838.375
|
||||
628286.375
|
||||
753114.25
|
||||
610243.3125
|
||||
627454.3125
|
||||
757834.5
|
||||
734318.5
|
||||
946036.25
|
||||
665812.5625
|
||||
622410.25
|
||||
643881.375
|
||||
723412.5625
|
||||
620958.9375
|
||||
923849.4375
|
||||
736022.625
|
||||
976675.1875
|
||||
718078.875
|
||||
653185.5
|
||||
701069.5
|
||||
560771.6875
|
||||
917033.25
|
||||
719322.875
|
||||
893176.4375
|
||||
575460.625
|
||||
941742.0
|
||||
933835.25
|
||||
730399.25
|
||||
759027.375
|
||||
724435.0
|
||||
910217.0
|
||||
735698.8125
|
||||
746246.9375
|
||||
816215.5
|
||||
797147.125
|
||||
674264.6875
|
||||
760731.375
|
||||
707053.625
|
||||
745105.25
|
||||
724230.5
|
||||
757323.3125
|
||||
735341.0
|
||||
717278.0
|
||||
746570.6875
|
||||
894829.375
|
||||
757152.875
|
||||
917033.25
|
||||
549899.8125
|
||||
1070398.25
|
||||
828893.625
|
||||
942082.875
|
||||
534733.6875
|
||||
530643.9375
|
||||
742838.8125
|
||||
737726.625
|
||||
741134.75
|
||||
753063.125
|
||||
724605.375
|
||||
731080.8125
|
||||
759879.375
|
||||
622969.75
|
||||
622066.5625
|
||||
543083.5625
|
||||
1022684.6875
|
||||
746332.125
|
||||
703901.125
|
||||
629328.75
|
||||
625730.3125
|
||||
703815.9375
|
||||
543236.9375
|
||||
743350.0
|
||||
820563.75
|
||||
539505.0625
|
||||
822608.625
|
||||
726548.0
|
||||
730586.625
|
||||
736908.6875
|
||||
681029.8125
|
||||
886019.375
|
||||
628647.125
|
||||
737845.9375
|
||||
752824.625
|
||||
754767.25
|
||||
936118.625
|
||||
566122.4375
|
||||
717278.0
|
||||
718982.0
|
||||
760390.625
|
||||
983340.9375
|
||||
730910.4375
|
||||
737726.625
|
||||
558420.0625
|
||||
716235.625
|
||||
561811.125
|
||||
710666.25
|
||||
761242.625
|
||||
762265.0625
|
||||
800776.75
|
||||
544787.625
|
||||
728183.9375
|
||||
1022684.6875
|
||||
759879.375
|
||||
738220.8125
|
||||
705042.875
|
||||
725798.25
|
||||
712268.0625
|
||||
766695.625
|
||||
532859.25
|
||||
732614.5
|
||||
1072085.25
|
||||
1051210.625
|
||||
736942.75
|
||||
825826.375
|
||||
766695.625
|
||||
537971.375
|
||||
734318.5
|
||||
657871.6875
|
||||
767240.875
|
||||
595298.75
|
||||
753693.625
|
||||
747951.0
|
||||
742838.8125
|
||||
754750.1875
|
||||
518715.5625
|
||||
774023.0625
|
||||
738527.5625
|
||||
718982.0
|
||||
533285.25
|
||||
727604.5625
|
||||
549899.8125
|
||||
639893.875
|
||||
677042.3125
|
||||
626448.875
|
||||
717278.0
|
||||
946002.1875
|
||||
778624.0
|
||||
743179.625
|
||||
718198.1875
|
||||
721026.875
|
||||
683057.625
|
||||
982449.0625
|
||||
742838.8125
|
||||
765673.1875
|
||||
841012.375
|
||||
558300.8125
|
||||
551603.875
|
||||
1031205.0
|
||||
706917.3125
|
||||
570518.875
|
||||
725798.25
|
||||
920441.375
|
||||
727672.75
|
||||
741134.75
|
||||
676340.75
|
||||
570348.5
|
||||
722185.625
|
||||
733125.6875
|
||||
597613.375
|
||||
744491.75
|
||||
696249.9375
|
||||
906519.1875
|
||||
721776.6875
|
||||
1299764.25
|
||||
735903.3125
|
||||
644665.25
|
||||
541209.125
|
||||
950091.9375
|
||||
720686.125
|
||||
997123.875
|
||||
961338.6875
|
||||
696249.9375
|
||||
708076.0625
|
||||
897907.9375
|
||||
736022.625
|
||||
919027.0
|
||||
906808.875
|
||||
703901.125
|
||||
651481.5
|
||||
720856.5
|
||||
748479.25
|
||||
1029500.9375
|
||||
727502.3125
|
||||
727502.3125
|
||||
736022.625
|
||||
772540.5
|
||||
716425.9375
|
||||
725798.25
|
||||
1139844.375
|
||||
654634.0
|
||||
968683.1875
|
||||
1203314.625
|
||||
656593.625
|
||||
741305.125
|
||||
573756.5625
|
||||
583980.9375
|
||||
736022.625
|
||||
729325.625
|
||||
578868.75
|
||||
663921.0625
|
||||
613992.25
|
||||
717278.0
|
||||
726070.875
|
||||
794471.75
|
||||
849683.125
|
||||
705349.625
|
||||
648260.8125
|
||||
722731.0
|
||||
961338.6875
|
||||
697831.8125
|
||||
669203.6875
|
||||
781827.625
|
||||
654634.0
|
||||
512580.96875
|
||||
721231.375
|
||||
722407.1875
|
||||
784639.3125
|
||||
757152.875
|
||||
934585.0
|
||||
727246.6875
|
||||
732273.6875
|
||||
766695.625
|
||||
690674.75
|
||||
768399.625
|
||||
766695.625
|
||||
727195.625
|
||||
802480.75
|
||||
1147512.625
|
||||
975823.125
|
||||
751359.125
|
||||
713529.0625
|
||||
745735.75
|
||||
578868.75
|
||||
642961.1875
|
||||
555011.9375
|
||||
688970.6875
|
||||
764991.5625
|
||||
738016.375
|
||||
696249.9375
|
||||
549899.8125
|
||||
549899.8125
|
||||
580402.375
|
||||
551603.875
|
||||
583980.9375
|
||||
715642.125
|
||||
708451.0
|
||||
744525.875
|
||||
776511.0
|
||||
816624.4375
|
||||
643642.8125
|
||||
604429.625
|
||||
534563.3125
|
||||
734318.5
|
||||
538823.4375
|
||||
760561.0
|
||||
712847.4375
|
||||
629158.375
|
||||
724366.875
|
||||
722901.375
|
||||
644665.25
|
||||
624878.25
|
||||
710291.375
|
||||
659831.375
|
||||
736022.625
|
||||
543697.0
|
||||
561828.1875
|
||||
580572.8125
|
||||
535347.125
|
||||
549559.0
|
||||
745122.25
|
||||
623174.1875
|
||||
798050.25
|
||||
646164.8125
|
||||
734318.5
|
||||
522242.96875
|
||||
639945.0
|
||||
553307.875
|
||||
556716.0
|
||||
540033.3125
|
||||
712336.25
|
||||
651481.5
|
||||
583980.9375
|
||||
756982.5
|
||||
624196.625
|
||||
672782.1875
|
||||
715573.9375
|
||||
1073826.25
|
||||
585565.6875
|
||||
595057.25
|
||||
620808.4375
|
||||
724435.0
|
||||
741305.125
|
||||
674486.25
|
||||
522242.96875
|
||||
754920.5625
|
||||
910029.5625
|
||||
750847.875
|
||||
559544.75
|
||||
825826.375
|
||||
786803.5
|
||||
732273.6875
|
||||
591700.3125
|
||||
562492.75
|
||||
922176.5625
|
||||
724094.1875
|
||||
757834.5
|
||||
1077595.25
|
||||
788848.375
|
||||
686605.0
|
||||
733602.875
|
||||
1009802.0625
|
||||
720686.125
|
||||
727502.3125
|
||||
1090847.0
|
||||
732392.9375
|
||||
530030.5
|
||||
740776.875
|
||||
751427.25
|
||||
675338.25
|
||||
735153.5
|
||||
769081.25
|
||||
1038021.25
|
||||
551603.875
|
||||
560635.375
|
||||
734318.5
|
||||
714426.4375
|
||||
530473.5625
|
||||
879714.375
|
||||
546474.625
|
||||
695786.9375
|
||||
737726.625
|
||||
540646.75
|
||||
649862.625
|
||||
538295.1875
|
||||
845474.125
|
||||
714892.3125
|
||||
1046030.25
|
||||
568184.3125
|
||||
644648.1875
|
||||
985826.0
|
||||
734744.5625
|
||||
725661.9375
|
||||
1041278.875
|
||||
1041278.875
|
||||
1099898.375
|
||||
771194.3125
|
||||
1002767.1875
|
||||
554807.5
|
||||
745991.3125
|
||||
731609.125
|
||||
736022.625
|
||||
892903.75
|
||||
733449.5
|
||||
663409.875
|
||||
736806.5
|
||||
782424.0625
|
||||
743588.625
|
||||
743588.625
|
||||
1152573.625
|
||||
721538.125
|
||||
709453.4375
|
||||
1077725.75
|
||||
769047.1875
|
||||
751785.125
|
||||
736976.875
|
||||
520981.96875
|
||||
713835.75
|
||||
712114.6875
|
||||
736124.875
|
||||
783242.0
|
||||
593574.75
|
||||
764718.875
|
|
58
train.py
Normal file
58
train.py
Normal file
@ -0,0 +1,58 @@
|
||||
import torch
|
||||
import torch.nn as nn
|
||||
import numpy as np
|
||||
import pandas as pd
|
||||
import sys
|
||||
import torch.nn.functional as F
|
||||
import pickle
|
||||
from torch.utils.data import TensorDataset, DataLoader
|
||||
|
||||
TRAIN_FILE_PATH = 'train/train.tsv'
|
||||
|
||||
|
||||
#prepare data methods
|
||||
def read_data_file(filepath):
|
||||
df = pd.read_csv(filepath, sep='\t', header=None, index_col=None)
|
||||
dataframe = df.iloc[:, [0,8,11]]
|
||||
dataframe.columns = ['price','biggy','type']
|
||||
#print(dataframe.size[0])
|
||||
for x in range(len(dataframe)):
|
||||
dataframe['biggy'].loc[x] = dataframe['biggy'].loc[x].replace(" ","")
|
||||
#such dumb solution, well, but at least it works
|
||||
dataframe['bias'] = 1
|
||||
dataframe['biggy'] = dataframe['biggy'].astype(float)
|
||||
return dataframe
|
||||
|
||||
|
||||
def dataframe_to_arrays(dataframe):
|
||||
dataframe1 = dataframe.copy(deep=True)
|
||||
dataframe1["type"] = dataframe1["type"].astype('category').cat.codes
|
||||
inputs_array = dataframe1[input_cols].to_numpy()
|
||||
targets_array = dataframe1[output_cols].to_numpy()
|
||||
return inputs_array, targets_array
|
||||
|
||||
data = read_data_file(TRAIN_FILE_PATH)
|
||||
input_cols = data.columns.values[1:]
|
||||
output_cols = data.columns.values[:1]
|
||||
|
||||
|
||||
|
||||
inputs_array_training, targets_array_training = dataframe_to_arrays(data)
|
||||
|
||||
inputs_training = torch.from_numpy(inputs_array_training).type(torch.float32)
|
||||
targets_training = torch.from_numpy(targets_array_training).type(torch.float32)
|
||||
|
||||
print(inputs_training)
|
||||
w = torch.tensor([7201.61492633873, 1,7201.500], requires_grad=True)
|
||||
learning_rate = torch.tensor(0.000000000005)
|
||||
print("training started")
|
||||
for i in range(10000):
|
||||
y_predicted = inputs_training @ w
|
||||
cost = torch.sum((y_predicted - targets_training) ** 2)
|
||||
cost.backward()
|
||||
with torch.no_grad():
|
||||
w -= learning_rate * w.grad
|
||||
w.requires_grad = True
|
||||
|
||||
print(w)
|
||||
pickle.dump(w, open('model.pkl', 'wb'))
|
Loading…
Reference in New Issue
Block a user